IndexError: too many indices for tensor of dimension 0

小咪咪 2022-12-18 04:57 226阅读 0赞

源代码：

  1. import torch
  2. import torchvision
  3. import torch.nn as nn
  4. from model import LeNet
  5. import torch.optim as optim
  6. import torchvision.transforms as transforms
  7. from torch.utils.data import DataLoader
  9. transform = transforms.Compose(
  10.     [transforms.ToTensor(),
  11.      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))
即可

细节！细节！细节！

发表评论

表情：
评论列表 （有 0 条评论，226人围观）

还没有评论，来说两句吧...

相关阅读