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.

Sign up for GitHub

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

Jump to bottom

Accelerator Refactor: Precision Plugins #5718

Merged
merged 14 commits into from
Jan 31, 2021
Merged

Accelerator Refactor: Precision Plugins #5718

merged 14 commits into from
Jan 31, 2021

Commits on Jan 31, 2021

  1. add basic accelerator class. 

    Co-Authored with @awaelchi
    @justusschock
    justusschock committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    a6e6e20 View commit details
    Browse the repository at this point in the history

  2. add basic trainign type plugin. 

    Co-Authored with @awaelchi
    @justusschock
    justusschock committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    3c5b0a6 View commit details
    Browse the repository at this point in the history

  3. pep8 

    Co-authored-by: @awaelchi
    @justusschock
    justusschock committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    5ab4b9e View commit details
    Browse the repository at this point in the history

  4. update copyright 

    Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
    @justusschock @awaelchli
    justusschock and awaelchli committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    ab4660d View commit details
    Browse the repository at this point in the history

  5. add apex_amp 

    Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
    @justusschock @awaelchli
    justusschock and awaelchli committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    1ade664 View commit details
    Browse the repository at this point in the history

  6. add mixed base class 

    Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
    @justusschock @awaelchli
    justusschock and awaelchli committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    bcd0e7a View commit details
    Browse the repository at this point in the history

  7. add native amp 

    Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
    @justusschock @awaelchli
    justusschock and awaelchli committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    a287567 View commit details
    Browse the repository at this point in the history

  8. add native amp sharded 

    Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
    @justusschock @awaelchli
    justusschock and awaelchli committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    1ff9b42 View commit details
    Browse the repository at this point in the history

  9. add tpu bfloat 

    Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
    @justusschock @awaelchli
    justusschock and awaelchli committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    1b82554 View commit details
    Browse the repository at this point in the history

  10. add inits 

    Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
    @justusschock @awaelchli
    justusschock and awaelchli committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    791fe19 View commit details
    Browse the repository at this point in the history

  11. Update precision_plugin.py

    @justusschock
    justusschock committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    94e0b28 View commit details
    Browse the repository at this point in the history

  12. Update pytorch_lightning/plugins/base_plugin.py 

    Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
    @justusschock @Borda
    justusschock and Borda committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    97eab50 View commit details
    Browse the repository at this point in the history

  13. fix imports

    @justusschock
    justusschock committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    de8fe1b View commit details
    Browse the repository at this point in the history

  14. Merge branch 'ref/precision_plugins' of github.com:PyTorchLightning/p… 

    …ytorch-lightning into ref/precision_plugins
    @justusschock
    justusschock committed Jan 31, 2021
    Configuration menu
    Copy the full SHA
    60445ad View commit details
    Browse the repository at this point in the history