torch_dataset=Data.TensorDataset(data_tensor=x,target_tensor=y)
报错
Traceback (most recent call last):
File "E:/python代码/pytorch_pandas_Test4/MiniBatch.py", line 10, in <module>
torch_dataset=Data.TensorDataset(data_tensor=x,target_tensor=y)
TypeError: __init__() got an unexpected keyword argument 'data_tensor'
新版把之前的data_tensor 和target_tensor去掉了,输入变成了可变参数
解决方法:直接将参数传入
# 原版使用方法
train_dataset = Data.TensorDataset(data_tensor=x, target_tensor=y)
# 新版使用方法
train_dataset = Data.TensorDataset(x,y)