pytorch: AttributeError: 'builtin_function_or_method' object,将dataloader返回值放到cuda再传回到网络结构中出现的问题及解决

在使用PyTorch时,作者在train.py中遇到将dataloader返回的tensor放入cuda时引发AttributeError。问题出在将tensor直接在dataloader返回后放入cuda。解决方案是将放tensor到cuda的操作移至网络模型内部处理,避免在数据加载器返回后立即进行。此方法解决了错误,使得代码能正常运行。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

将dataloader返回值放到cuda再传回到网络结构中出现的问题及解决

问题描述

题主主要编写了三个文件,datasets.py,models.py,train.py
datasets.py返回五个tensor,四个img和1个label
models.py设计了三个类,一个特征提取类——放在网络的前面,一个分类网络——放到网络的后面,还有一个类,负责将特征提取类,融合策略层以及分类网络整合在一起。

但是,在train.py中,我尝试将loader中的tensor返回,出现了
AttributeError: ‘builtin_function_or_method’ object这种错误,就很奇怪,不用cuda的话,就不会有这个错误。

for down, left, right, upper, label in train_loader:
	down, left, right, upper 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值