torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Added model wrappers for DeepFM and DLRM

Open SSYernar opened this issue 5 months ago • 2 comments

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

SSYernar avatar Jun 18 '25 20:06 SSYernar