benchmark
0c44e4d1 - Increase the batch size from mobilenet training

Commit
4 years ago
Increase the batch size from mobilenet training Summary: Increase the batch size from 32 to 96 according to the original paper's evaluation setup. https://github.com/pytorch/benchmark/issues/155 Test Plan: ``` python test.py -k "test_mobilenet_v2_train_cuda" python test.py -k "test_mobilenet_v2_train_cpu" python test.py -k "test_mobilenet_v2_quantized_qat_train_cuda" python test.py -k "test_mobilenet_v2_quantized_qat_train_cpu" ``` Reviewers: Subscribers: Tasks: Tags:
Author
Committer
Parents
Loading