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

Training improvements #17

Merged
merged 10 commits into from
Aug 23, 2024
Merged

Training improvements #17

merged 10 commits into from
Aug 23, 2024

Commits on Aug 12, 2024

  1. improves training interface

    @evanhanders
    evanhanders committed Aug 12, 2024
    Configuration menu
    Copy the full SHA
    6c5f4ed View commit details
    Browse the repository at this point in the history

Commits on Aug 14, 2024

  1. A bunch of small changes to make training smoother; need to debug and… 

    … type-check.
    @evanhanders
    evanhanders committed Aug 14, 2024
    Configuration menu
    Copy the full SHA
    1c089cf View commit details
    Browse the repository at this point in the history

Commits on Aug 21, 2024

  1. New SIIT sampling protocol, Adds optimizer_kwargs to training_args

    @evanhanders
    evanhanders committed Aug 21, 2024
    Configuration menu
    Copy the full SHA
    97c18d3 View commit details
    Browse the repository at this point in the history

Commits on Aug 22, 2024

  1. more training fixups (args, etc)

    @evanhanders
    evanhanders committed Aug 22, 2024
    Configuration menu
    Copy the full SHA
    5e5eaa0 View commit details
    Browse the repository at this point in the history

  2. merge

    @evanhanders
    evanhanders committed Aug 22, 2024
    Configuration menu
    Copy the full SHA
    855807a View commit details
    Browse the repository at this point in the history

  3. fixes typing errors

    @evanhanders
    evanhanders committed Aug 22, 2024
    Configuration menu
    Copy the full SHA
    7734db5 View commit details
    Browse the repository at this point in the history

  4. Defaults: use_single_step = True, betas = (0.9, 0.9); adds optimzier_… 

    …cls to training_args
    @evanhanders
    evanhanders committed Aug 22, 2024
    Configuration menu
    Copy the full SHA
    18e467e View commit details
    Browse the repository at this point in the history

Commits on Aug 23, 2024

  1. cleans up PR per feedback

    @evanhanders
    evanhanders committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    0d1b24f View commit details
    Browse the repository at this point in the history

  2. reverts some indexing in IITBehaviorModelPair that caused circuits-be… 

    …nch to break
    @evanhanders
    evanhanders committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    313309e View commit details
    Browse the repository at this point in the history

  3. bug fix and adds back in one .to(device)

    @evanhanders
    evanhanders committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    e4eba0a View commit details
    Browse the repository at this point in the history