在python编写时,遇到了这个问题,我的代码中,是在遍历的时候出现的问题
for i, (inputs, heatmaps_targets, gts) in enumerate(trainDataLoader):
#for i, values in enumerate(trainDataLoader):
inputs = Variable(inputs).cuda()
主要是因为enumerate(trainDataLoader)就输出一个值,如果你这么改就可以了
for i, values in enumerate(trainDataLoader):
values是list类型。如果values有多个值,你可以这样改:
for i, values in enumerate(trainDataLoader):
print(values[0])
print(values[1])
就可以了,记得点赞,谢谢小伙伴~