mxnet 和pytorch比较

最新推荐文章于 2024-09-21 22:34:33 发布

AI算法网奇

最新推荐文章于 2024-09-21 22:34:33 发布

阅读量6.2k

点赞数 1

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/115726421

版权

pytorch知识宝典专栏收录该内容

503 篇文章 228 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文探讨了mxnet和pytorch在构建神经网络时的不同之处，mxnet采用链式结构，而pytorch则支持列表结构。在mxnet中，symbol的特征维度打印成为一个问题，且设计网络时不需要指定输入的channel数，而在pytorch中则必须明确输入通道数。

摘要由CSDN通过智能技术生成

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=

了解本专栏