提要
最近我在学习莫烦的神经网络的批训练时,遇到了一个问题,就是在创建dataset时报__init__() got an unexpected keyword argument 'data_tensor',我找了很久都没有有效的结果。但是我偶然尝试了一下结果成功了,下面就列出调试过程
原始情况
原始代码
x = torch.linspace(1, 10, 10) # x data (torch tensor)
y = torch.linspace(10, 1, 10) # y data (torch tensor)
# 先转换成 torch 能识别的 Dataset
torch_dataset = Data.TensorDataset(data_tensor=x, target_tensor=y

在学习Pytorch神经网络批训练时,遇到'got an unexpected keyword argument'错误。从原始代码报错开始,经过初步调试如去除data_tensor和target_tensor,发现错误反而加重。进一步排查发现是多线程问题,需添加开始入口。最终成功运行,但反思问题根源可能涉及更多因素。
最低0.47元/天 解锁文章
2万+





