实验室也没有可以使用的机器,只能在自己的电脑上跑,本机性能堪忧,在跑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)
Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐