赞
踩
最近我在学习莫烦的神经网络的批训练时,遇到了一个问题,就是在创建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 能识别的 Data
赞
踩
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。