个人记录

pytorch训练过程中GPU利用率低

问题在训练人脸数据集MS1M时，采用pytorch的ImageFolder对原始的图像进行的读取。由于人脸数据小，且量大，导致GPU很快训练完成，但是IO却很慢，从而拖垮了整个训练时间。解决方法以上问题的根本原因在于pytorch没有自己的数据格式，像TF的TFrecorde，mx的rec文件...
2020-07-05 16:50 阅读数 181
技术
下载桌面版
GitHub
Gitee
SourceForge
百度网盘（提取码：draw）
云服务器优惠
华为云优惠券
腾讯云优惠券
阿里云优惠券
Vultr优惠券
站点信息
问题反馈
邮箱：[email protected]
吐槽一下
QQ群：766591547
关注微信
©2020-2024 ioDraw All rights reserved, Privacy Policy