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

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_29007291/article/details/84782909

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

---

做了一个性能分析，关于训练网络时的时间消耗问题。
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上运行