Hi, I am using pytorch to build CNN model. But I am facing some problem with data generation. Here is my data shape:
X_train = (-1, 1, 169999, 200)
X_test = (-1, 1, 3000, 200)
y_train = (169999, 200)
y_test =  (3000, 200)
train = torch.utils.data.TensorDataset(X_train, y_train)
test = torch.utils.data.TensorDataset(X_test, y_test)
But after that I am getting this error:
assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors)
AssertionError