benchmark
0954aa61 - Fix yolov3 model training to original batch size (#630)

Commit
4 years ago
Fix yolov3 model training to original batch size (#630) Summary: This PR improves the quality of yolov3 model. - Use the original train batch size of 16 (https://github.com/ultralytics/yolov3/blob/master/train.py#L447) - Increase eval size to 32 to saturate GPU - Package input files to a single `coco128.tar.gz` file and manage it with Git LFS - Disable logs that are not used Train profile: ![image](https://user-images.githubusercontent.com/502017/145507541-b3876250-b2e4-4b91-83d6-e06619c6f5ff.png) Eval profile: ![image](https://user-images.githubusercontent.com/502017/145508290-dbbedb42-8ac3-4a80-af19-69707c00bfb9.png) Pull Request resolved: https://github.com/pytorch/benchmark/pull/630 Reviewed By: aaronenyeshi Differential Revision: D33032976 Pulled By: xuzhao9 fbshipit-source-id: dcb0a7b216ea2d4707022430580ebf75ac668a0f
Author
Parents
Loading