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

[PyTorch] Fix autocast deprecation warnings #1277

Open
wants to merge 5 commits into
base: main
Choose a base branch
from
Open

[PyTorch] Fix autocast deprecation warnings #1277

wants to merge 5 commits into from

Commits on Oct 23, 2024

  1. fix autocast related deprecation warning 

    Signed-off-by: Xin Yao <[email protected]>
    @yaox12
    yaox12 committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    d08b273 View commit details
    Browse the repository at this point in the history

  2. add docstring 

    Signed-off-by: Xin Yao <[email protected]>
    @yaox12
    yaox12 committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    3adc513 View commit details
    Browse the repository at this point in the history

  3. fix kwargs for torch.amp.autocast 

    Signed-off-by: Xin Yao <[email protected]>
    @yaox12
    yaox12 committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    e70dfb1 View commit details
    Browse the repository at this point in the history

  4. check torch version inside functions 

    Signed-off-by: Xin Yao <[email protected]>
    @yaox12
    yaox12 committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    39a9fe6 View commit details
    Browse the repository at this point in the history

  5. check torch version inside functions 

    Signed-off-by: Xin Yao <[email protected]>
    @yaox12
    yaox12 committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    d162789 View commit details
    Browse the repository at this point in the history