Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[DeepSpeed] Remove partitioning of model in ZeRO 3 #10655

Merged
merged 11 commits into from
Dec 17, 2021
Merged

Commits on Nov 20, 2021

  1. deepspeed does not require an optimizer for inference, refactor test …

    …to use parametrization
    SeanNaren committed Nov 20, 2021
    Configuration menu
    Copy the full SHA
    5b1c596 View commit details
    Browse the repository at this point in the history
  2. Update CI

    SeanNaren committed Nov 20, 2021
    Configuration menu
    Copy the full SHA
    0959818 View commit details
    Browse the repository at this point in the history
  3. Add CHANGELOG.md

    SeanNaren committed Nov 20, 2021
    Configuration menu
    Copy the full SHA
    5297ce2 View commit details
    Browse the repository at this point in the history
  4. Remove partition module to see what breaks

    SeanNaren committed Nov 20, 2021
    Configuration menu
    Copy the full SHA
    6413ef3 View commit details
    Browse the repository at this point in the history

Commits on Nov 22, 2021

  1. Update CHANGELOG.md

    Sean Naren authored Nov 22, 2021
    Configuration menu
    Copy the full SHA
    a1e3650 View commit details
    Browse the repository at this point in the history
  2. Remove param

    SeanNaren committed Nov 22, 2021
    Configuration menu
    Copy the full SHA
    041a7ed View commit details
    Browse the repository at this point in the history
  3. Remove optimizer change

    SeanNaren committed Nov 22, 2021
    Configuration menu
    Copy the full SHA
    3d2e82d View commit details
    Browse the repository at this point in the history
  4. Fix test

    SeanNaren committed Nov 22, 2021
    Configuration menu
    Copy the full SHA
    3701fde View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    ccfa6e1 View commit details
    Browse the repository at this point in the history

Commits on Dec 17, 2021

  1. Merge branch 'master' into fix/update_deepspeed

    # Conflicts:
    #	CHANGELOG.md
    #	pytorch_lightning/plugins/training_type/deepspeed.py
    #	tests/plugins/test_deepspeed_plugin.py
    SeanNaren committed Dec 17, 2021
    Configuration menu
    Copy the full SHA
    1ea8db8 View commit details
    Browse the repository at this point in the history
  2. Add todo to broken case

    SeanNaren committed Dec 17, 2021
    Configuration menu
    Copy the full SHA
    21526cd View commit details
    Browse the repository at this point in the history