Open 993526915 opened 4 months ago
为什么在代码运行的时候 dataloader_train = DataLoader(train_dataset, batch_size=args.batch_size, num_workers=args.workers,drop_last=True)这段代码没法讲dataset变成dataloader,dataloader无法通过迭代获取数据
pytorch版本问题,需要用到pytorch1.10版本,现已解决
为什么在代码运行的时候 dataloader_train = DataLoader(train_dataset, batch_size=args.batch_size, num_workers=args.workers,drop_last=True)这段代码没法讲dataset变成dataloader,dataloader无法通过迭代获取数据