Fix torchrec_dlrm model and enable fbcode test (#1428)
Summary:
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1428
Fix a bug in the train test, the train pipeline requires multiple batches of inputs, but the iterator has only 1 batch.
Fix the iterator to use `cycle()` which returns infinite batches of data.
Also, enable the torchrec_dlrm model running in fbcode.
Inductor still doesn't work for this model, stack backtrace is attached.
Reviewed By: colin2328
Differential Revision: D43397292
fbshipit-source-id: e7907d7883d4f15f2117baf2b04a0cef6f5b5fcc
