关注
码龄
粉丝数
原力等级 --
被采纳
被点赞
采纳率
猫七^
2022-03-10 16:58
采纳率: 75%
浏览 218
首页
人工智能
已结题
faster rcnn提高训练速度
pytorch
深度学习
我用的抓取检测模型里面用到了fastrcnn 结合vmrd和抓取检测,训练vmrd数据集。但训练过程很慢,增大batch_size训练速度也没有提高多少。
这个图是我的训练过程,感觉训练完需要一天,实在太慢
这张图是训练用到的显存,用的GPU1,24446的显存只用了9232
应该修改哪些参数提高训练速度呢?想尽可能的提高显存的利用率
收起
写回答
好问题
0
提建议
关注问题
微信扫一扫
点击复制链接
分享
邀请回答
编辑
收藏
删除
收藏
举报
1
条回答
默认
最新
关注
码龄
粉丝数
原力等级 --
被采纳
被点赞
采纳率
影醉阏轩窗
2022-03-11 10:35
关注
数据存储优化:S3(读取http)、直接到内存
数据处理欧化:dali/ffcv
模型优化:全部上gpu,用空间换时间。也可以考虑DDP多卡训练。
可能有效:opencv线程、并行线程等设置
anchor部分优化:比如for改为维度计算,可以参考yolov5
。。。
看你这张图gpu利用率那么低,大概率卡在CPU阶段了。如果没有额外的卡,先提高gpu利用率吧。
...
本回答被题主选为最佳回答
, 对您是否有帮助呢?
本回答被专家选为最佳回答
, 对您是否有帮助呢?
本回答被题主和专家选为最佳回答
, 对您是否有帮助呢?
解决
无用
评论
打赏
微信扫一扫
点击复制链接
分享
举报
编辑记录
评论
按下Enter换行,Ctrl+Enter发表内容
查看更多回答(0条)
向“C知道”追问
报告相同问题?
提交
关注问题
向专家提问
向AI提问
付费问答(悬赏)服务下线公告
◇ 用户帮助中心
◇ 新手如何提问
◇ 奖惩公告
问题事件
关注
码龄
粉丝数
原力等级 --
被采纳
被点赞
采纳率
系统已结题
3月19日
关注
码龄
粉丝数
原力等级 --
被采纳
被点赞
采纳率
已采纳回答
3月11日
关注
码龄
粉丝数
原力等级 --
被采纳
被点赞
采纳率
修改了问题
3月11日
关注
码龄
粉丝数
原力等级 --
被采纳
被点赞
采纳率
创建了问题
3月10日