pytorch 深度学习, CPU预处理时间大于GPU网络处理时间、数据增强

背景:想知道训练网络时,时间花在了哪.


做了一个性能分析,关于训练网络时的时间消耗问题。
1.电脑: i7-6700,1070 ti
2.tiny-YOLO_v1(9层) ,
3.预处理包括随机平移,缩放,Hue,Saturation等

结果

batch-size = 18时, 一个batch平均用时:
image 加载和预处理 : 0.4849s;
网络前向+损失计算+后传用时:0.04308s

结论

1.网络过小时, CPU和GPU用时比例不合理,可能需要更大的网络才能合理利用GPU
2.需要处理器的性能很强才行
3.预处理部分不知道能不能在GPU上运行

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值