Fix the train arch of DLRM

Commit

4 years ago

Fix the train arch of DLRM Use the training arch from the DLRM paper: Source: https://arxiv.org/pdf/1906.00091.pdf