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

Add Additional Logging Metrics #1331

Merged

Quentin-Anthony merged 2 commits into main from qanthony/improve-logging

Dec 19, 2024

Member

Quentin-Anthony commented Dec 18, 2024

Logs MFU/HFU if the user passes neox_args.peak_theoretical_tflops
Log tokens_per_sec, iters_per_sec to wandb, comet, and tensorboard
Add ability to manually specify wandb experiment name

Quentin-Anthony added 2 commits

December 18, 2024 11:16


          add wandb_run_name arg

5ae53db


          add mfu, hfu, tokens_per_sec, and iters_per_sec logging

97e57d5

Quentin-Anthony merged commit 29080c3 into main

1 of 4 checks passed

Quentin-Anthony deleted the qanthony/improve-logging branch

December 19, 2024 21:51

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

Labels

None yet