Skip to content

v0.2.1

Choose a tag to compare

View all tags
@thetianshuhuang thetianshuhuang released this 04 Jan 20:27
· 57 commits to main since this release
v0.2.1
682b4fb
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

What's changed: DX Improvements by @thetianshuhuang in #21

Improvements to the developer experience:

  • Better logging & error messages
  • Only write high-level log messages from rank 0 in multi-GPU training to avoid cluttering the log
  • Better looking outputs using rich trees for nrdk inspect
  • Log gradient norm in NRDKLightningModule

Other fixes:

  • Only write checkpoints.yaml from rank 0 to prevent concurrency bugs
  • Freeze models loaded with framework.load_model by default
  • Fix @cache @torch.compiler.disable ordering

Full Changelog: v0.2.0...v0.2.1

Contributors

thetianshuhuang
Assets 2
Loading