csmiler
csmiler
Summary: Deprecate CPU `rowwise_adagrad_with_counter` kernel Differential Revision: D52589984
Summary: Deprecate CPU|GPU `rowwise_weighted_adagrad` kernels Differential Revision: D52575349
Differential Revision: D56643198
Summary: X-link: https://github.com/facebookresearch/FBGEMM/pull/335 Following D63909559, support checkpointing `iter` for Adam/Partial_rowwise_adam/Lamb/Partial_rowwise_lamb Differential Revision: D63964848
Summary: `self.step` is used for logging purposes only and is not properly checkpointed/reloaded. Instead, `iter` should be used as it's reloaded from external https://fburl.com/code/48caflz4 Differential Revision: D63616621