Skip to content
This repository has been archived by the owner on Mar 19, 2024. It is now read-only.

Auto infer optimizer when using zero and convergence test of zero2 + pytorch amp + activation checkpointing #178

Closed
wants to merge 1 commit into from

Conversation

prigoyal
Copy link
Contributor

@prigoyal prigoyal commented Feb 8, 2021

Summary:
auto inferring the zero optimizer setting which basically moves the parameters to "base_optimizer".

now users just need to set use_zero=True .

Differential Revision: D26236748

…pytorch amp + activation checkpointing

Summary:
auto inferring the zero optimizer setting which basically moves the parameters to "base_optimizer".

now users just need to set `use_zero=True` .

Differential Revision: D26236748

fbshipit-source-id: bd6fecea869892e1c336f8d25aad6e1a0bad061b
@facebook-github-bot facebook-github-bot added CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported labels Feb 8, 2021
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D26236748

facebook-github-bot pushed a commit that referenced this pull request Feb 9, 2021
…pytorch amp + activation checkpointing (#178)

Summary:
Pull Request resolved: #178

auto inferring the zero optimizer setting which basically moves the parameters to "base_optimizer".

now users just need to set `use_zero=True` .

Reviewed By: min-xu-ai

Differential Revision: D26236748

fbshipit-source-id: 326db1040a457391735632071fe718d6c6071947
facebook-github-bot pushed a commit that referenced this pull request Aug 12, 2021
Summary:
In prevision for the work on the optimiser group enhancement for FSDP:
- These tests will allow to catch some accidental issues that arise if refactoring is needed.
- These tests add an important use case to our list of integration tests: fine tuning.

Pull Request resolved: fairinternal/ssl_scaling#178

Reviewed By: prigoyal

Differential Revision: D30233695

Pulled By: QuentinDuval

fbshipit-source-id: ca67bc5f65328976298c3b09b17048dd4bfde934
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants