torchrec
torchrec copied to clipboard
add get_fused_optimizers
Summary: Since we no longer rely on DistributedModelParallel (for composability piece), we need an alternative way of getting the fused optimizer. get_fused_optimizer implements this, logically it's the same as the recursive call in DMP
Differential Revision: D37505229
This pull request was exported from Phabricator. Differential Revision: D37505229