遇到这种情况 ——深度学习
- 首先检查自己的数据下载部分,是不是出现重复下载。这里的重复下载是指在训练源数据的部分,单独一个函数train(train_loader,target_loader)中存在数据ids与data下载,不要将其放置在真正的循环训练中(因为本人为了容易理解,将其放入循环中了导致程序运行很慢);
- 其次检查DataLoader中的num_workers=?,我也不清楚为什么这里设置为8,会使程序运行加快。总之,将其内部参数设置为pin_memory=True, num_workers=8。
- 我看到有的链接说是因为文件IO操作导致的,也就是log_interval的值需要适当变大,这个我不太理解到底还需要在哪里需要修改。
目前我的程序比之前快了一些,但是还不是很快。