多进程导致的内存不够用了吧

将数据集读取的进程修改为

trainset = torchvision.datasets.CIFAR100(root=args.data_path, train=True, download=False, transform=transform_train)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=args.batch_size, shuffle=True, num_workers=0)
testset = torchvision.datasets.CIFAR100(root=args.data_path, train=False, download=False, transform=transform_test)
testloader = torch.utils.data.DataLoader(testset, batch_size=args.batch_size, shuffle=False, num_workers=0)

  num_workers = 0 既可

相关文章:

  • 2021-12-29
  • 2021-08-04
  • 2022-12-23
  • 2021-09-11
  • 2022-12-23
  • 2022-12-23
  • 2023-03-17
  • 2021-06-19
猜你喜欢
  • 2022-12-23
  • 2021-12-26
  • 2022-01-03
  • 2022-12-23
  • 2021-07-12
相关资源
相似解决方案