torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard
verified publisher icon pytorch

enable gradient accumulation in SDD

Open arsatis opened this issue 2 months ago • 1 comments

Summary: Context: gradient accumulation is still not available in TorchRec, especially for the SDD pipeline which is being used by many recommendation models.

This diff: implements a UI enabling gradient accumulation for SDD.

Differential Revision: D84915986

arsatis avatar Oct 17 '25 14:10 arsatis

@arsatis has exported this pull request. If you are a Meta employee, you can view the originating Diff in D84915986.

meta-codesync[bot] avatar Oct 17 '25 14:10 meta-codesync[bot]