pytorch - Pytorch List 对象没有属性“to”

Question

我是 Pytorch 的新手。我正在使用预训练模型（RESNET-50）。我正在尝试使用 cocoapi 在 MS-COCO 数据集上训练这个模型。

我已经在火车加载器中加载了我的数据集图像和注释。

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)

我得到以下输出：

类“火炬。张量”

类“列表”

火炬尺寸（[64, 3, 224, 224]）

4

我得到了这个错误：我认为这是因为标签不是张量？