Andy_2259 2022-01-08 14:30 采纳率: 0%
浏览 228
已结题

visdom报错:TypeError: Object of type Tensor is not JSON serializable

利用visdom进行绘图,使用CPU跑的时候可以正常运行,使用GPU跑的时候会报错。
绘图语句:
viz.line(np.array(train_accs), X=np.arange(epoch+1), win='tacc{}'.format(task_id),opts={'title': 'Task {}: train accuracy'.format(task_id)})

使用GPU跑的时候报错为:TypeError: Object of type Tensor is not JSON serializable

使用CPU跑时候的Y数据为[tensor(0.1058)]
使用GPU跑时候的Y数据为[tensor(0.0167, device='cuda:0')]
是因为GPU跑时候数据不对导致无法画图吗?

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 1月16日
    • 创建了问题 1月8日