模型训练速度过慢,GPU利用率低

遇到这种情况 ——深度学习

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

目前我的程序比之前快了一些,但是还不是很快。

你可能感兴趣的:(python,Pycharm,深度学习,迁移学习)