使用visdom可视化pytorch训练时报错：TypeError: Object of type Tensor is not JSON serializable

使用visdom可视化pytorch训练时报错：TypeError: Object of type Tensor is not JSON serializable

今天学习使用visdom对pytorch训练过程中的loss进行可视化。部分代码如下：

loss0 = criterion0(f_y0, f_out_y0, w0) + 0.01 * l2_regularization0
loss1 = criterion1(f_y1, f_out_y1, w1) + 0.01 * l2_regularization1

#更新窗口图像，训练中loss的可视化
viz.line([[float(loss0), float(loss1)]], [epoch], win='train', update='append')

然后运行的时候就报错了：

TypeError: Object of type Tensor is not JSON serializable

问题分析：

• loss变量是tensor类型
• 报错的大概意思也就是tensor这个类型不行
• 程序给出的推荐方法是：var.detach().numpy()，也就是把tensor类型转换成numpy类型。然而尝试之后依然报错！
• 最后尝试直接将tensor类型转换成浮点数，该方法成功！ 也就是把之前的代码改成如下形式：

viz.line([[float(loss0), float(loss1)]], [epoch], win='train', update='append')

总结：

该问题主要是对于visdom使用过程中的数据类型没搞清楚。浮点数是可用选项。

另附：
Github上的回答