torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Add helper to apply overlapped optimizer to module

Open lequytra opened this issue 3 years ago • 1 comments

Summary: Since applying an overlapped optimizer to all EBCs and ECs in a model is a common use case, creating a helper function apply_overlapped_optimizer_to_module doing just that for convenience.

Differential Revision: D39580955

lequytra avatar Sep 16 '22 20:09 lequytra

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

facebook-github-bot avatar Sep 16 '22 20:09 facebook-github-bot