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