批训练错误之got an unexpected keyword...

在学习Pytorch神经网络批训练时,遇到'got an unexpected keyword argument'错误。从原始代码报错开始,经过初步调试如去除data_tensor和target_tensor,发现错误反而加重。进一步排查发现是多线程问题,需添加开始入口。最终成功运行,但反思问题根源可能涉及更多因素。
摘要由CSDN通过智能技术生成

提要

       最近我在学习莫烦的神经网络的批训练时,遇到了一个问题,就是在创建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
  • 18
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值