背景:想知道训练网络时,时间花在了哪.
做了一个性能分析,关于训练网络时的时间消耗问题。
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上运行