PyTorch-Tutorial icon indicating copy to clipboard operation
PyTorch-Tutorial copied to clipboard

Update 305_batch_train.ipynb

Open YumiGo opened this issue 3 years ago • 0 comments

Modify argument in TensorDataset

Error code: Data.TensorDataset(data_tensor=x, target_tensor=y)

Fixed code: Data.TensorDataset(x, y)

YumiGo avatar May 12 '22 15:05 YumiGo