Fix the train arch of DLRM (#580)
Summary:
Use the training arch from the DLRM paper:
Source: https://arxiv.org/pdf/1906.00091.pdf
Pull Request resolved: https://github.com/pytorch/benchmark/pull/580
Reviewed By: xuzhao9
Differential Revision: D32668551
Pulled By: aaronenyeshi
fbshipit-source-id: 379bb2cc3a108a9f414c340938a5a9f3cdbb3cfe
