Lee Yao Yang
Results
2
comments of
Lee Yao Yang
For the TensorDataSet NameError, I found that adding this line solves the issue `from torch.utils.data import TensorDataset`
I would recommend upgrading numpy as well.