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

Add PyTorch AMP check #7917

Merged
merged 11 commits into from
May 22, 2022
Merged

Add PyTorch AMP check #7917

merged 11 commits into from
May 22, 2022

Commits on May 21, 2022

  1. Add PyTorch AMP check

    @glenn-jocher
    glenn-jocher committed May 21, 2022
    Configuration menu
    Copy the full SHA
    1c8204f View commit details
    Browse the repository at this point in the history

  2. [pre-commit.ci] auto fixes from pre-commit.com hooks 

    for more information, see https://pre-commit.ci
    @pre-commit-ci
    pre-commit-ci[bot] committed May 21, 2022
    Configuration menu
    Copy the full SHA
    2f92c3c View commit details
    Browse the repository at this point in the history

  3. Cleanup

    @glenn-jocher
    glenn-jocher committed May 21, 2022
    Configuration menu
    Copy the full SHA
    6ad990a View commit details
    Browse the repository at this point in the history

  4. Cleanup

    @glenn-jocher
    glenn-jocher committed May 21, 2022
    Configuration menu
    Copy the full SHA
    3ccf8cb View commit details
    Browse the repository at this point in the history

  5. Cleanup

    @glenn-jocher
    glenn-jocher committed May 21, 2022
    Configuration menu
    Copy the full SHA
    34e9cca View commit details
    Browse the repository at this point in the history

  6. Robust for DDP

    @glenn-jocher
    glenn-jocher committed May 21, 2022
    Configuration menu
    Copy the full SHA
    d71f39c View commit details
    Browse the repository at this point in the history

Commits on May 22, 2022

  1. Fixes

    @glenn-jocher
    glenn-jocher committed May 22, 2022
    Configuration menu
    Copy the full SHA
    4dbe6aa View commit details
    Browse the repository at this point in the history

  2. Add amp enabled boolean to check_train_batch_size

    @glenn-jocher
    glenn-jocher committed May 22, 2022
    Configuration menu
    Copy the full SHA
    0dcecc3 View commit details
    Browse the repository at this point in the history

  3. Simplify

    @glenn-jocher
    glenn-jocher committed May 22, 2022
    Configuration menu
    Copy the full SHA
    78613e0 View commit details
    Browse the repository at this point in the history

  4. space to prefix

    @glenn-jocher
    glenn-jocher committed May 22, 2022
    Configuration menu
    Copy the full SHA
    ca11e37 View commit details
    Browse the repository at this point in the history

  5. Merge branch 'master' into amp_check

    @glenn-jocher
    glenn-jocher committed May 22, 2022
    Configuration menu
    Copy the full SHA
    7106e21 View commit details
    Browse the repository at this point in the history