Я получаю ошибку атрибута, так как объект «int 'не имеет атрибута» к' ⇐ Python

[Anonymous]

Я пишу код Python в ноутбуке Kaggle для классификации изображений. В учебной части я получаю ошибку < /p>

AttributeError Traceback (most recent call last)
in
5 test_loss = 0.0
6 for images,label in enumerate(train_loader):
----> 7 images,label = images.to(cuda),label.to(cuda)
8 optimizer.zero_grad()
9

AttributeError: 'int' object has no attribute 'to'
< /code>

Это следующий код, (я даю только 2 части, пожалуйста, скажите, вам нужно больше) < /p>

train_loader = torch.utils.data.DataLoader(train_data,batch_size = 128,num_workers =0,shuffle =True)
test_loader = torch.utils.data.DataLoader(test_data,batch_size = 64,num_workers =0,shuffle =False)

epoch = 10

for e in range(epoch):
train_loss = 0.0
test_loss = 0.0
for images,label in enumerate(train_loader):
images,label = images.to(cuda),label.to(cuda)
optimizer.zero_grad()

output = model(images)
_,predict = torch.max(output.data, 1)
loss = criterion(output,labels)
loss.backward()
optimizer.step()

train_loss += loss.item()
train_size += label.size(0)
train_success += (predict==label).sum().item()

print("train_accuracy is {.2f}".format(100*(train_success/train_size)) )


Подробнее здесь: https://stackoverflow.com/questions/616 ... tribute-to