torchrec
torchrec copied to clipboard
Added model wrappers for DeepFM and DLRM
Summary:
- Added model wrappers for DeepFM and DLRM. The wrapper will take ModelInput as an only parameter in the forward method.
- Added the unit tests to cover the models' wrappers
Differential Revision: D76916471