IndexError: too many indices for tensor of dimension 0

源代码:

import torch
import torchvision
import torch.nn as nn
from model import LeNet
import torch.optim as optim
import torchvision.transforms as transforms
from torch.utils.data import DataLoader

transform = transforms.Compose(
    [transforms.ToTensor(),
     transforms.Normalize(0.5, 0.5,0.5), (0.5,0.5,0.5)])

以上是部分源代码,为展示完整代码。出现标题所示错误的原因是:
transforms.Normalize(0.5, 0.5,0.5), (0.5,0.5,0.5)
改为
transforms.Normalize((0.5, 0.5,0.5), (0.5,0.5,0.5))
即可

细节!细节!细节!

你可能感兴趣的:(python,bug)