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.

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

Make NTK computation independent module #120

Closed
wants to merge 8 commits into from

Conversation

KonstiNik
Copy link
Member

@KonstiNik KonstiNik commented May 10, 2024

Make the NTK computation its own module.

Tasks:

  • Create module carrying out the comptutation
  • Remove ntk computation from all models
  • Adapt the API used in tests
  • Adapt the API of the example notebooks

This is relevant regarding the new recorders, which do not include NTK calculation anymore.
Moreover, with this approach, one can design and apply training processes more flexibly.

@KonstiNik
Copy link
Member Author

Completely covered in 30c1e70

@KonstiNik KonstiNik closed this Nov 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant