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

feat: Added support of AMP to all PyTorch training scripts #604

Merged
merged 9 commits into from
Nov 10, 2021
Merged

feat: Added support of AMP to all PyTorch training scripts #604

merged 9 commits into from
Nov 10, 2021

Commits on Nov 10, 2021

  1. refactor: Renamed the model argument to arch

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    42f5789 View commit details
    Browse the repository at this point in the history

  2. refactor: Removed epoch index logging

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    f0986e7 View commit details
    Browse the repository at this point in the history

  3. feat: Added dataset hash logging

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    0251045 View commit details
    Browse the repository at this point in the history

  4. feat: Added elements to the payload logged on W&B at each epoch

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    e17dbd2 View commit details
    Browse the repository at this point in the history

  5. refactor: Removed unused arguments

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    5a6be5f View commit details
    Browse the repository at this point in the history

  6. feat: Cleaned sample plotting

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    6edca88 View commit details
    Browse the repository at this point in the history

  7. feat: Added AMP support to text detection training

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    0f0fcaf View commit details
    Browse the repository at this point in the history

  8. feat: Added AMP for pytorch trainings

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    1af2e73 View commit details
    Browse the repository at this point in the history

  9. fix: Fixed compatibility with FP16

    @fg-mindee
    fg-mindee committed Nov 10, 2021
    Configuration menu
    Copy the full SHA
    a4de6c4 View commit details
    Browse the repository at this point in the history