skyundead
skyundead
dataloader = data_.DataLoader(dataset, \ batch_size=1, \ shuffle=True, \ # pin_memory=True, num_workers=opt.num_workers)
> 但是我发现如果我们将批处理大小设置为> 1,则会引发异常错误。 haha,me too。if batch_size = 2,it can‘t work。
> 我已经更新了代码,只需要torchvision和pytorch,不需要安装cupy,nms等第三方代码了,理论上可以在windows下直接跑的 出现了ModuleNotFoundError: No module named 'resource'的问题,直接注释点 不影响训练吧?
> 好像只能加载第2个预训练模型,第一个和第3个一加载就卡死了 我的都可以加载