Add custom model fwd in train pipelines#2324

sarckk wants to merge 1 commit intopytorch:mainpytorch/torchrec:mainfrom

sarckk:export-D61564467sarckk/torchrec:export-D61564467Copy head branch name to clipboard