原版代码:
torch_dataset = Data.TensorDataset(data_tensor=x, target_tensor=y)
loader = Data.DataLoader(
dataset=torch_dataset, # torch TensorDataset format
batch_size=BATCH_SIZE, # mini batch size
shuffle=True, # random shuffle for training
num_workers=2, # subprocesses for loading data
)
运行报错:
TypeError Traceback (most recent call last)
<ipython-input-19-5bb67537d9eb> in <module>
----> 1 torch_dataset = Data.TensorDataset(data_tensor=x, target_tensor=y)
2 # 代码修改为:参考https://blog.youkuaiyun.com/idwtwt/article/details/86767634
3 # torch_dataset = Data.TensorDataset(x, y)
4 loader = Data.DataLoader(
5 dataset=torch_dataset, # torch TensorDataset format
TypeError: __init__() got an unexpected keyword argumen