Skip to content

Conversation

@n0kovo
Copy link

@n0kovo n0kovo commented Feb 2, 2025

Added support for logging metrics to Weights & Biases.
Created a new reporting module for it, so other stuff like TensorBoard etc. could go there too. Let me know if anything needs changing.
Thanks for an awesome project!

@armbues
Copy link
Owner

armbues commented Feb 6, 2025

Awesome contribution, thank you!

From a quick glance this looks good to me, but I wanted to check the specific changes in trainer.py once again before merging.

One (cosmetic) change that I'd ask for: could you move wand.py to the utils folder?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants