Fix hardcoded shape in low_mem_dropout benchmark (#2475)
Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/2475
Reviewed By: htyu
Differential Revision: D63653081
Pulled By: xuzhao9
fbshipit-source-id: 8d840986779b6124cbccc2425c24e2b892d55ce4
