Add a test to ensure feature_processor weights are checkpointed

Open xush6528 opened this issue 3 years ago • 2 comments

Summary:

Problem

Identified for the CMF model on TorchRec, its feature processors were not checkpointed.

Add a unit test to guard its behavior

Reviewed By: ge0405

Differential Revision: D39518283

Sep 19 '22 17:09 xush6528

This pull request was exported from Phabricator. Differential Revision: D39518283

Sep 19 '22 17:09 facebook-github-bot

This pull request was exported from Phabricator. Differential Revision: D39518283

Sep 19 '22 18:09 facebook-github-bot