torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Added Optimizer configuration that supports optimizer type, learning rate, momentum, and weight decay configurations.

Open SSYernar opened this issue 5 months ago • 2 comments

Summary: This commit introduces enhancements to the optimizer configuration in TorchRec. It now supports specifying the optimizer type, learning rate, momentum, and weight decay. These changes provide more flexibility and control over the training process, allowing users to fine-tune their models with different optimization strategies and hyperparameters.

Differential Revision: D76559261

SSYernar avatar Jun 13 '25 02:06 SSYernar