benchmark
af847a96 - Increase the default batch size of alexnet (#919)

Commit
3 years ago
Increase the default batch size of alexnet (#919) Summary: The default batch size for alexnet is too small and is not a proper workload for GPU. Some performance issues (e.g., https://github.com/pytorch/pytorch/issues/76970) only affects small batch size inputs and I believe they are insignificant in real settings. This PR increases the default batch size for alexnet on GPU. Although this PR will also increase CPU batch size, we are working on a PR that uses separate batch sizes for different devices, which will use small batch size for CPU, and large batch size for GPU. Pull Request resolved: https://github.com/pytorch/benchmark/pull/919 Reviewed By: jansel Differential Revision: D36549692 Pulled By: xuzhao9 fbshipit-source-id: 03552c1f5617670797b7ea9069be926293da4f34
Author
Parents
Loading