pytorch一些问题

• 运行时报warning：
  UserWarning: test_data has been renamed datawarnings.warn("test_data has been renamed data")
UserWarning: test_labels has been renamed targets warnings.warn("test_labels has been renamed targets")
  后来发现是载入数据集的属性应该是data和target，而不是之前写的test_data,test_labels,改完之后即可。

test_x = torch.unsqueeze(test_data.data, dim=1).type(torch.FloatTensor)[:2000]/255.   # shape from (2000, 28, 28) to (2000, 1, 28, 28), value in range(0,1)
test_y = test_data.targets[:2000]