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

Implemented the loss NTK calculation #109

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

Implemented the loss NTK calculation #109

wants to merge 35 commits into from

Commits on Jan 12, 2024

  1. starting out

    Marc Sauter committed Jan 12, 2024
    Configuration menu
    Copy the full SHA
    076d0df View commit details
    Browse the repository at this point in the history

Commits on Jan 18, 2024

  1. Renaming file

    @ma-sauter
    ma-sauter committed Jan 18, 2024
    Configuration menu
    Copy the full SHA
    930cac8 View commit details
    Browse the repository at this point in the history

  2. analysis file should be done

    @ma-sauter
    ma-sauter committed Jan 18, 2024
    Configuration menu
    Copy the full SHA
    580194a View commit details
    Browse the repository at this point in the history

  3. quick fix

    @ma-sauter
    ma-sauter committed Jan 18, 2024
    Configuration menu
    Copy the full SHA
    768fc13 View commit details
    Browse the repository at this point in the history

Commits on Jan 19, 2024

  1. Included loss ntk, eigval and entropy in jax recorder

    @ma-sauter
    ma-sauter committed Jan 19, 2024
    Configuration menu
    Copy the full SHA
    c802010 View commit details
    Browse the repository at this point in the history

  2. Updated recorder instantiation test

    @ma-sauter
    ma-sauter committed Jan 19, 2024
    Configuration menu
    Copy the full SHA
    17fd83a View commit details
    Browse the repository at this point in the history

  3. Bugfixing

    @ma-sauter
    ma-sauter committed Jan 19, 2024
    Configuration menu
    Copy the full SHA
    e88292f View commit details
    Browse the repository at this point in the history

  4. Stand vor Kino

    Marc Sauter committed Jan 19, 2024
    Configuration menu
    Copy the full SHA
    4eb8362 View commit details
    Browse the repository at this point in the history

Commits on Jan 22, 2024

  1. First thing state that might be working

    @ma-sauter
    ma-sauter committed Jan 22, 2024
    Configuration menu
    Copy the full SHA
    fb1da2e View commit details
    Browse the repository at this point in the history

  2. writing test

    Marc Sauter committed Jan 22, 2024
    Configuration menu
    Copy the full SHA
    d22bb68 View commit details
    Browse the repository at this point in the history

Commits on Jan 30, 2024

  1. Included vmap_axes

    @ma-sauter
    ma-sauter committed Jan 30, 2024
    Configuration menu
    Copy the full SHA
    e141dd8 View commit details
    Browse the repository at this point in the history

  2. Working on calculating loss derivatives to calculate loss ntk comparison

    @ma-sauter
    ma-sauter committed Jan 30, 2024
    Configuration menu
    Copy the full SHA
    22cfc90 View commit details
    Browse the repository at this point in the history

Commits on Feb 4, 2024

  1. Calculation and test should work now

    @ma-sauter
    ma-sauter committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    5f9b6bf View commit details
    Browse the repository at this point in the history

  2. some linting updates

    @ma-sauter
    ma-sauter committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    131d5cb View commit details
    Browse the repository at this point in the history

  3. Fixed tests

    @ma-sauter
    ma-sauter committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    5b244ff View commit details
    Browse the repository at this point in the history

Commits on Feb 6, 2024

  1. Some modifications to simplify the loss ntk test code

    @ma-sauter
    ma-sauter committed Feb 6, 2024
    Configuration menu
    Copy the full SHA
    3a06e75 View commit details
    Browse the repository at this point in the history

  2. Class renaming to follow convention

    @ma-sauter
    ma-sauter committed Feb 6, 2024
    Configuration menu
    Copy the full SHA
    001e46e View commit details
    Browse the repository at this point in the history

  3. added reshape and unshape methods in the loss_ntk_calculation

    @ma-sauter
    ma-sauter committed Feb 6, 2024
    Configuration menu
    Copy the full SHA
    fd99b77 View commit details
    Browse the repository at this point in the history

Commits on Feb 20, 2024

  1. quicksave

    Marc Sauter committed Feb 20, 2024
    Configuration menu
    Copy the full SHA
    0ef059f View commit details
    Browse the repository at this point in the history

  2. Quick save

    @ma-sauter
    ma-sauter committed Feb 20, 2024
    Configuration menu
    Copy the full SHA
    98c6b30 View commit details
    Browse the repository at this point in the history

  3. Added test for eigenvalues, precision is still only e-4

    @ma-sauter
    ma-sauter committed Feb 20, 2024
    Configuration menu
    Copy the full SHA
    cd5644a View commit details
    Browse the repository at this point in the history

  4. Added some docstrings

    @ma-sauter
    ma-sauter committed Feb 20, 2024
    Configuration menu
    Copy the full SHA
    b4fe246 View commit details
    Browse the repository at this point in the history

  5. More docstrings

    @ma-sauter
    ma-sauter committed Feb 20, 2024
    Configuration menu
    Copy the full SHA
    bf7a401 View commit details
    Browse the repository at this point in the history

  6. Black formatting

    @ma-sauter
    ma-sauter committed Feb 20, 2024
    Configuration menu
    Copy the full SHA
    43cdcb9 View commit details
    Browse the repository at this point in the history

  7. More docstrings

    @ma-sauter
    ma-sauter committed Feb 20, 2024
    Configuration menu
    Copy the full SHA
    fd8626f View commit details
    Browse the repository at this point in the history

Commits on Feb 21, 2024

  1. Some PR modifications

    @ma-sauter
    ma-sauter committed Feb 21, 2024
    Configuration menu
    Copy the full SHA
    b6ebb8d View commit details
    Browse the repository at this point in the history

Commits on Feb 26, 2024

  1. fixing PR comment

    Marc Sauter committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    d17e514 View commit details
    Browse the repository at this point in the history

  2. requirements change

    Marc Sauter committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    eb49043 View commit details
    Browse the repository at this point in the history

  3. Black formatter changes

    Marc Sauter committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    53c548d View commit details
    Browse the repository at this point in the history

  4. changed recorder to use the use_loss_ntk flag

    Marc Sauter committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    1dac434 View commit details
    Browse the repository at this point in the history

  5. Change recorder test for new flag

    Marc Sauter committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    422eceb View commit details
    Browse the repository at this point in the history

  6. removed unneccesary CNN model from loss_ntk calculation test

    Marc Sauter committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    b5e8170 View commit details
    Browse the repository at this point in the history

  7. Started integration test for loss_ntk_calculation

    Marc Sauter committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    594f4cb View commit details
    Browse the repository at this point in the history

Commits on Feb 27, 2024

  1. Implemented integration test

    Marc Sauter committed Feb 27, 2024
    Configuration menu
    Copy the full SHA
    b55fc30 View commit details
    Browse the repository at this point in the history

  2. isort

    Marc Sauter committed Feb 27, 2024
    Configuration menu
    Copy the full SHA
    080099e View commit details
    Browse the repository at this point in the history