Skip to content

TRAK v0.3.0
Compare
Choose a tag to compare
@kristian-georgiev kristian-georgiev released this 03 Nov 18:55
· 4 commits to main since this release
v0.3.0
76b13ca
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

0.3.0 by @kristian-georgiev and @AlaaKhaddaj in #50

  • Added support for large models and datasets (ChunkedCudaProjector, (much) faster scoring by removing I/O bottlenecks)
  • Allow taking gradients with respect to a selected set of parameter groups (e.g., only wrt last layer)
  • black codestyle
  • bug fixes

Contributors

AlaaKhaddaj and kristian-georgiev
Assets 2
Loading