Я использую простой поезд l oop для регрессионного задания. Чтобы убедиться, что значения истинности регрессии совпадают с теми, что я ожидал на тренировке l oop, я решил построить каждый пакет данных. Тем не менее, я вижу, что когда я конвертирую тензор загрузчика данных в массив 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()