小编Sam*_* H.的帖子

PyTorch Dataloader - 枚举时列表不可调用错误

当迭代 PyTorch 数据加载器时,例如

# define dataset, dataloader
train_data = datasets.ImageFolder(data_dir + '/train', transform=train_transforms)
test_data = datasets.ImageFolder(data_dir + '/test', transform=test_transforms)
trainloader = torch.utils.data.DataLoader(train_data, batch_size=64, shuffle=True)
testloader = torch.utils.data.DataLoader(test_data, batch_size=64)

# define model, optimizer, loss
# not included - irrelevant to the question

for ii, (inputs, labels) in enumerate(trainloader):

    # Move input and label tensors to the GPU
    inputs, labels = inputs.to(device), labels.to(device)

    start = time.time()

    outputs = model.forward(inputs)
    loss = criterion(outputs, labels)
    loss.backward()
Run Code Online (Sandbox Code Playgroud)

TypeError: 'list' object is not callable在这条线上得到了 …

pytorch

2
推荐指数
1
解决办法
4293
查看次数

标签 统计

pytorch ×1