利用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跑时候数据不对导致无法画图吗?