-
Notifications
You must be signed in to change notification settings - Fork 338
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
[feat] Extend aim.ext.tensorboard_tracker.run.Run
to allow stdout logging and system stats and parameter logging
#2671
[feat] Extend aim.ext.tensorboard_tracker.run.Run
to allow stdout logging and system stats and parameter logging
#2671
Conversation
…ure terminal logs
…g occurs when enabled.
aim.ext.tensorboard_tracker.run.Run
to allow stdout logging and system stats and parameter loggingaim.ext.tensorboard_tracker.run.Run
to allow stdout logging and system stats and parameter logging
Everything looks good to me. Please just sign the CLA and add a CHANGELOG entry. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Implementation looks good!
Have some thoughts about merging the BasicRun
and Run
classes of SDK; as the only purpose for the separation was the tb sync functionality.
But we can leave it for now.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great job! 🙌
Please sign the CLA, and also there's a conflict that needs to be fixed before merging.
@alansaul could you please sign the CLA, so we can merge the enhancement? |
Apologies for the delay, I've merged upstream and signed the CLA, and fixed the changelog |
It looks like you might have some randomly failing tests as I don't believe I touched anything relating to the failing tests. |
…ogging and system stats and parameter logging (#2671) # Conflicts: # CHANGELOG.md
Implemented the simple approach to resolve #2670 by simply inheriting from
aim.sdk.run.Run
rather thanaim.sdk.run.BasicRun
, and we get the behaviour for free.Note: the default arguments for
system_tracking_interval
,log_system_params
andcapture_terminal_logs
are different from the defaults provided byaim.sdk.run.Run
to maintain backwards compatability, and by default not do any additional logging.I have also added a bunch of tests to check the extended behaviour works (default no capture, and that capturing each component works as expected). Perhaps there is an easier implementation for grabbing the metrics of interest, I might need some guidance as I am new to this project. I couldn't find anywhere where the logging behaviour of
aim.sdk.run.Run
is captured to copy from. Let me know if you need any changes/think these tests should be removed.Resolves #2670