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

Some improvements / bug fixes #476

Open
wants to merge 6 commits into
base: master
Choose a base branch
from
Open

Some improvements / bug fixes #476

wants to merge 6 commits into from

Commits on Apr 23, 2024

  1. Change callback population to preserve progress bar

    @mitchelldehaven
    mitchelldehaven committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    379bee1 View commit details
    Browse the repository at this point in the history

  2. Add early stopping to prevent training for too long and overfitting

    @mitchelldehaven
    mitchelldehaven committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    555e8aa View commit details
    Browse the repository at this point in the history

  3. Only save best model based on validation loss

    @mitchelldehaven
    mitchelldehaven committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    d8655a1 View commit details
    Browse the repository at this point in the history

  4. Put test set generation in 'on_validation_end', otherwise it gets ran… 

    … for each validation batch
    @mitchelldehaven
    mitchelldehaven committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    e6af4c6 View commit details
    Browse the repository at this point in the history

  5. Absolute value needs to be taken before max, this results in warning … 

    …messages due to amplitude issues
    @mitchelldehaven
    mitchelldehaven committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    de98f64 View commit details
    Browse the repository at this point in the history

  6. Force ops without half precision support in float32 context, allowing… 

    … fp16 training
    @mitchelldehaven
    mitchelldehaven committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    8eda759 View commit details
    Browse the repository at this point in the history