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

Custom loss trainer #48

Merged
merged 3 commits into from
Aug 24, 2023
Merged

Custom loss trainer #48

merged 3 commits into from
Aug 24, 2023

Commits on Aug 17, 2023

  1. allowing for custom loss in Trainer

    @rchan26
    rchan26 committed Aug 17, 2023
    Configuration menu
    Copy the full SHA
    a76734e View commit details
    Browse the repository at this point in the history

Commits on Aug 18, 2023

  1. loosen accelerate dependency for torch=1.9.0

    @rchan26
    rchan26 committed Aug 18, 2023
    Configuration menu
    Copy the full SHA
    e3f14fc View commit details
    Browse the repository at this point in the history

  2. add return_outputs arg to custom compute_loss when overriding Trainer… 

    … class
    @rchan26
    rchan26 committed Aug 18, 2023
    Configuration menu
    Copy the full SHA
    4f20c9f View commit details
    Browse the repository at this point in the history