【发布时间】:2019-03-21 14:50:54
【问题描述】:
尝试从以下 repo 从 github 运行 pytorch 代码
https://github.com/jalola/improved-wgan-pytorch
错误出现在文件 gan_train.py 中的以下语句
writer.add_scalar('data/disc_cost', disc_cost, iteration)
报错如下
File "/home/anaconda/new_anaconda_3.5/lib/python3.6/site-packages/tensorboardX/writer.py", line 272, in add_scalar
self.file_writer.add_summary(scalar(tag, scalar_value), global_step)
File "/home/anaconda/new_anaconda_3.5/lib/python3.6/site-packages/tensorboardX/summary.py", line 86, in scalar
scalar = makenp(scalar)
File "/home/anaconda/new_anaconda_3.5/lib/python3.6/site-packages/tensorboardX/x2num.py", line 15, in makenp
return pytorch_np(x, modality)
File "/home/anaconda/new_anaconda_3.5/lib/python3.6/site-packages/tensorboardX/x2num.py", line 24, in pytorch_np
if isinstance(x, torch.autograd.variable.Variable):
AttributeError: 'function' object has no attribute 'Variable'
任何想法,我对pytorch不是很精通。 tensorboard API 似乎有问题。
非常感谢。
山
【问题讨论】:
-
嘿,你解决问题了吗?如果您找到了解决方案,如果您能分享它,那就太好了!
标签: python-3.x pytorch