pytorch中的网络结构怎样可视化:
方法1:使用
model_path = '/home/fuxueping/sdb/PycharmProjects/mxnetTopytorch/se_best/model'
epoch = 29
#model_path路径/home/fuxueping/sdb/PycharmProjects/mxnetTopytorch/se_best中保存的mxnet的网络结构和模型参数(.json,.param)
sym, arg_params, aux_params = mx.model.load_checkpoint(model_path, epoch)
mx.viz.plot_network(sym, title='alexnet', save_format='jpg', hide_weights=True).view()
具体参考:https://blog.csdn.net/sunshine_in_moon/article/details/53033431
方法2:Netron
在github上进行下载:https://github.com/lutzroeder/Netron
或者直接使用:
pip install netron
在终端直接输入:
netron
就可以将网络结构文件拖入进行使用了(我这块使用在mxnet的json文件上)