实验室也没有可以使用的机器,只能在自己的电脑上跑,本机性能堪忧,在跑num_workers=2的数据都吃不消
test_dataloader = D.DataLoader(test_datasets,BATCH,True,num_workers=2)
改成:(删掉num_workers=2)
test_dataloader = D.DataLoader(test_datasets,BATCH,True)
本文讨论了在本地计算机上运行 DataLoader 时遇到的性能瓶颈问题,并提供了解决方案,即通过调整 DataLoader 的参数来减轻资源压力。
实验室也没有可以使用的机器,只能在自己的电脑上跑,本机性能堪忧,在跑num_workers=2的数据都吃不消
test_dataloader = D.DataLoader(test_datasets,BATCH,True,num_workers=2)
改成:(删掉num_workers=2)
test_dataloader = D.DataLoader(test_datasets,BATCH,True)
1226
1万+
3万+
1219

被折叠的 条评论
为什么被折叠?