问题描述
题主主要编写了三个文件,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 = down.cuda(

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

被折叠的 条评论
为什么被折叠?



