Skip to content
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

Migrate from wandb to tensorboard #122

Merged
merged 9 commits into from
Aug 3, 2024
Merged

Conversation

ziw-liu
Copy link
Collaborator

@ziw-liu ziw-liu commented Aug 2, 2024

Fix #116.

Example logs:

/hpc/projects/intracellular_dashboard/viral-sensor/infection_classification/models/test_tb

@mattersoflight
Copy link
Member

@ziw-liu thanks for these improvements! I read through the code and tried demo_fit.py. Training of losses, metrics, and images looks good. But, the model graph was not logged. Please add the code to do so.

Copy link
Member

@mattersoflight mattersoflight left a comment

Choose a reason for hiding this comment

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

works well

@mattersoflight mattersoflight merged commit 0ba38c7 into main Aug 3, 2024
4 checks passed
@ziw-liu ziw-liu deleted the wandb-to-tensorboard branch August 5, 2024 16:21
edyoshikun pushed a commit that referenced this pull request Aug 7, 2024
* wip: use lightning's tensorboard logger instead of wandb

* private logging methods

* log center slice only

* fix tensor cloning

* only log metrics on epoch

* add simple demo training script

* fix flaky test

* log graph + profiling

* switch to simple profiler

---------

Co-authored-by: Shalin Mehta <[email protected]>
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.

switching to tensorboard logger for organelle phenotyping
2 participants