在pytorch下使用tensorboardX的方法及踩过的坑

环境说明:win7,python3.6.5,cuda9.0

版本说明:使用pytorch1.0.1和tensorboard1.9.0,tensorflow-gpu1.9.0

在使用tensorboard之前需要首先安装tensorflow,我安装的是GPU版本的。

然后参考两个入门

基础版:https://www.jianshu.com/p/46eb3004beca

增强版:https://blog.csdn.net/qq_36653505/article/details/84727253

上面的代码可以跑通。

踩过的坑:

主要是两个,(1)logdir的路径最好是详细路径,是event文件所在的路径

(2)开头使用的pytorch0.4.0,但是在对model进行可视化时,总是报OperatorExportTypes的错误,通过查阅源码发现是torch的版本不支持OperatorExportTypes,所以将0.4.0版本的torch卸载改为1.0.1版本的torch,错误解决。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值