PyTorch
关注
分享
扫一扫
'Themis'
这个作者很懒，什么都没留下…
展开
-
PyTorch模型结构可视化方法总结（未完结）
总体来说有几种方法：netron、 1. netron（个人最喜欢） 具体方法：pytorch存下来的.pth文件转为.onnx文件，直接拖入netron网页就可以看 优点：画出来比较细致，布局效果好，直观 2. tensorboard 缺点：加上tensorboard中关于summarywriter的代码后，好像会让显存增加不少，而且第一张卡的显存特别多 ...原创 2020-05-12 15:50:42 · 2478 阅读 · 0 评论 -
pytorch自学入门记录
前提： 接触过Keras、mxnet等深度学习框架 自学资料： 根据知乎回答，参考别人学习的顺序 官方tutorial（60min），英文，中文翻译版本 比较靠谱的介绍博客：https://blog.youkuaiyun.com/u014380165/article/category/9271909 ...原创 2019-09-25 17:46:31 · 234 阅读 · 0 评论