2019年11月22日
pytorch模型可视化，torchviz，tensorboardX，文本方式
摘要： torchviz生成一个pdf，pdf怎样命名还不知道，或许只能默认命名。 注意，torchviz我试了下将模型和伪造输入移到cuda上就会报错，所以这种可视化直接在cpu运行上即可。 torchviz看梯度类型更清楚，tensorboard看数据流形状更清楚。 还有tensorboardX方式，两 阅读全文
posted @ 2019-11-22 09:43 永远怀着一颗学习的心 阅读(1224) 评论(0) 推荐(0) 编辑