Fix the train batch size of alexnet. (#575)
Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/575
Reviewed By: aaronenyeshi
Differential Revision: D32605952
Pulled By: xuzhao9
fbshipit-source-id: f3be036fc2afced0518361bb117b1b4446b860b1