赞
踩
实验室也没有可以使用的机器,只能在自己的电脑上跑,本机性能堪忧,在跑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)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。