Skip to content

Add auto torch optimizer parameter logging convenience functionality#33

Merged
wli51 merged 3 commits into
WayScience:mainfrom
wli51:auto-log-optimizer
Jul 21, 2026
Merged

Add auto torch optimizer parameter logging convenience functionality#33
wli51 merged 3 commits into
WayScience:mainfrom
wli51:auto-log-optimizer

Conversation

@wli51

@wli51 wli51 commented Jun 10, 2026

Copy link
Copy Markdown
Collaborator

enhance reproducibility and keep better record of training

…to enhance reproducibility and keep better record
@MattsonCam
MattsonCam self-requested a review July 17, 2026 21:21

@MattsonCam MattsonCam left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is a nice, bite-sized pr. Good job @wli51 !

Comment thread src/virtual_stain_flow/vsf_logging/MlflowLogger.py
wli51 added 2 commits July 21, 2026 16:47
…re functionalities out of the main logger module to prevent logger module from swelling indefinitely and facilitate testing.
@wli51

wli51 commented Jul 21, 2026

Copy link
Copy Markdown
Collaborator Author

Just realized all the cumulative additions made MlflowLogger.py a pretty big module. On top of adding tests did a small scale refactor moving out auto-logging related functionalities outside of the main logger to reduce module size and facilitate modular testing.

@wli51

wli51 commented Jul 21, 2026

Copy link
Copy Markdown
Collaborator Author

Thanks @MattsonCam for the review. Merging now!

@wli51
wli51 merged commit fcb32a9 into WayScience:main Jul 21, 2026
1 check passed
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