benchmark
3970f85b - Improve quality of pytorch_stargan (#648)

Commit
4 years ago
Improve quality of pytorch_stargan (#648) Summary: Default train batch size: 16 Source: https://github.com/yunjey/stargan/blob/94dd002e93a2863d9b987a937b85925b80f7a19f/main.py#L73 Added data prefetch and eval batch size support. Train profile (bs=16): ![image](https://user-images.githubusercontent.com/502017/146257343-91e9fae4-7810-4c80-a7c8-793077e7c16c.png) Eval profile (bs=16): ![image](https://user-images.githubusercontent.com/502017/145923896-5050a5a5-16c0-45fb-991e-a8caaf202e65.png) Pull Request resolved: https://github.com/pytorch/benchmark/pull/648 Reviewed By: aaronenyeshi Differential Revision: D33138537 Pulled By: xuzhao9 fbshipit-source-id: 1583cbdff3eb9f38c4a162bbd279b61d100d72f4
Author
Parents
Loading