美文网首页
深度学习各个部分数据大小抓取方法

深度学习各个部分数据大小抓取方法

作者: CPinging | 来源:发表于2021-01-07 14:58 被阅读0次

一、使用Summary方案

效果如图:

image.png

第一列为Layer名称,第二列为输出数据的维度,第三列为weight的数量。

具体方法:

  • 1 pip3 install torchsummary

  • 2 from torchsummary import summary

  • 3 然后在训练后添加summary这句话,后面传入net定义的模型,并传入dataset的维度即可打印出来;

...
net.train()
summary(net, (3, 32, 32))
...

相关文章

网友评论

      本文标题:深度学习各个部分数据大小抓取方法

      本文链接:https://www.haomeiwen.com/subject/hhczoktx.html