Update the rowwise adagrad optimizer to leverage optimizer state offloading, v3
Summary:
- Add compile-time flag to enable working with optimizer offloading
- Update the rowwise adagrad optimizer to leverage optimizer state offloading if the optimizer is TBE SSD, and if the user specified it
Differential Revision: D74827718
Deploy Preview for pytorch-fbgemm-docs ready!
| Name | Link |
|---|---|
| Latest commit | 03f77b87c00e7baaa8a597cd5e49c042bd44b2c8 |
| Latest deploy log | https://app.netlify.com/projects/pytorch-fbgemm-docs/deploys/682f888d50eda10008c49d99 |
| Deploy Preview | https://deploy-preview-4133--pytorch-fbgemm-docs.netlify.app |
| Preview on mobile | Toggle QR Code...Use your smartphone camera to open QR code link. |
To edit notification comments on pull requests, go to your Netlify project configuration.
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718
This pull request was exported from Phabricator. Differential Revision: D74827718