mxnet 和pytorch比较

mxnet网络是链式结构，pytorch可以是列表结构

引发的问题：mxnet symbol如何打印特征维度？

mxnet设计网络是，不用输入网络输入channel，

pytorch需要输入通道数。

mxnet:

  

   

    

     

    
    

     

        num_classes = config.emb_size
     
    
   

    

     

    
    

     

       bn_mom = config.bn_mom
     
    
   

    

     

    
    

     

       workspace = config.workspace
     
    
   

    

     

    
    

     

       data = mx.symbol.Variable(name="data")  # 224
     
    
   

    

     

    
    

     

       data = data - 127.5
     
    
   

    

     

    
    

     

       data = data * 0.0078125
     
    
   

    

     

    
    

     

       fc_type = config.net_output
     
    
   

    

     

    
    

     

       bf = int(32 * config.net_multiplier)
     
    
   

    

     

    
    

     
 if config.net_input == 0:
     
    
   

    

     

    
    

     

       conv_1 = Conv(data, num_filter=bf, kernel=(3, 3), pad=(1, 1), stride=(2, 2), name="conv_1")  # 224/112
     
    
   

    

     

    
    

     
 else:
     
    
   

    

     

    
    

     

       conv_1 = Conv(data, num_filter=bf, kernel=(3, 3), pad=(1, 1), stride=(1, 1), name="conv_1")  # 224/112
     
    
   

    

     

    
    

     

       conv_2_dw = Conv(conv_1, num_group=bf, num_filter=bf, kernel=(3, 3), pad=(1, 1), stride=(1
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/115726421