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

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

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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值