eat_pytorch_in_20_days
eat_pytorch_in_20_days copied to clipboard
第一节代码运行报错!
dl_train = DataLoader(TensorDataset(torch.tensor(x_train).float(),torch.tensor(y_train).float()), shuffle = True, batch_size = 8) dl_valid = DataLoader(TensorDataset(torch.tensor(x_test).float(),torch.tensor(y_test).float()), shuffle = False, batch_size = 8)
AssertionError Traceback (most recent call last)
~\anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataset.py in init(self, *tensors) 34 35 def init(self, *tensors): ---> 36 assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors) 37 self.tensors = tensors 38
AssertionError: