在出现错误的地方加上：
with torch.no_grad():
test_loader = init_data_loader(X_test, y_test, torch.device(‘cpu‘), batch_size, num_workers=0) print("Starting inference ...") result = [] model.eval() valid_loss = 0 with torch.no_grad(): for batch_x, batch_y in tqdm(test_loader): batch_x, batch_y = batch_x.to(device), batch_y.to(device) output = model(batch_x) result.extend(output[:, 0, 0]) loss = torch.sqrt(criterion(output, batch_y)) valid_loss += loss valid_loss /= len(train_loader) print("Done!")
原文：https://www.cnblogs.com/dyc99/p/12664126.html