tensorboard-pytorch has been renamed to tensorboardX. The old version(0.7.1) would crash when I run Chapter03/03_atari_gan.py without cuda.
When calling writer.add_image("fake", vutils.make_grid(gen_output_v.data[:64], normalize=True), iter_no), it raises an error which says 'input tensor should be one of numpy.ndarray, torch.cuda.FloatTensor, torch.FloatTensor'. But without cuda, the tensor type is torch.Tensor.
The latest version of tensorboardX has fixed this issue.
tensorboard-pytorch has been renamed to tensorboardX. The old version(0.7.1) would crash when I run Chapter03/03_atari_gan.py without cuda.
When calling
writer.add_image("fake", vutils.make_grid(gen_output_v.data[:64], normalize=True), iter_no)
, it raises an error which says'input tensor should be one of numpy.ndarray, torch.cuda.FloatTensor, torch.FloatTensor'
. But without cuda, the tensor type istorch.Tensor
.The latest version of tensorboardX has fixed this issue.