我正在使用一个简单的训练循环来执行回归任务。为了确保回归的真实值与我在训练循环中的预期相同,我决定绘制每批数据。但是,我看到当我将数据加载器的张量转换为 numpy 数组并绘制它时,它受到了干扰。我正在使用 myTensor.data.cpu().numpy() 进行转换。
我的代码如下:
train_ds = TensorDataset(x_train, y_train)
train_dl = DataLoader(train_ds, batch_size = 32, shuffle = True, num_workers = 0, drop_last = True)
for epoch in range(epochs):
model.train()
for i, (x, y) in enumerate(train_dl):
x = x.cuda()
y = y.cuda()
yy = y.data.cpu().numpy()
pyplot.plot(yy[0: 32, 0])
pyplot.show()