dataloader问题 - Githubissues

mrwu-mac / DIFNet

[CVPR 2022] This repository is for the paper ``DIFNet: Boosting Visual Information Flow for Image Captioning'' .

BSD 3-Clause "New" or "Revised" License

20 stars 7 forks source link

dataloader问题 #16

Open 993526915 opened 7 months ago

993526915 commented 7 months ago

为什么在代码运行的时候 dataloader_train = DataLoader(train_dataset, batch_size=args.batch_size, num_workers=args.workers,drop_last=True)这段代码没法讲dataset变成dataloader，dataloader无法通过迭代获取数据

993526915 commented 7 months ago

1713103463351

993526915 commented 7 months ago

pytorch版本问题，需要用到pytorch1.10版本，现已解决