Объект списка Pytorch не имеет атрибута 'to'

Question

Я новичок в Pytorch. Я использую предварительно обученную модель (RE SNET -50). И я пытаюсь обучить эту модель на наборе данных MS-COCO с использованием cocoapi.

Я загрузил изображения и аннотации набора данных в загрузчик поездов.

for images,labels in train_loader:
    print(type(images))
    print(type(labels))
    print(images.size())
    print(images.dim())
    steps+=1
    images, labels = images.to(device), labels.to(device)

я получил следующий вывод:

класс 'torch.Tensor'

класс 'list'

torch.Size ([64, 3, 224, 224])

4

И я получил эту ошибку: я думаю, что это потому, что метки не является тензором?