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

Wandb #80

Merged
merged 7 commits into from
Feb 6, 2024
Merged

Wandb #80

merged 7 commits into from
Feb 6, 2024

Conversation

JLrumberger
Copy link
Collaborator

@JLrumberger JLrumberger commented Feb 1, 2024

What is the purpose of this PR?

In this PR we add a residual unet with mirror padding as a new backbone architecture and wandb logging of validation metrics to the training pipeline.

How did you implement your changes
I replaced the tensorboard logging functionality in ModelBuilder and PromixNaive with wandb logging and changed the vanilla unet architecture to use resnet style skip connections in the conv blocks. In addition I threw out a couple of tests from the deepcell PanopticNet, because they threw assertions without me changing any code or dependencies and I couldn't resolve them within a couple of hours.

Remaining issues

None

Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Copy link
Member

@ngreenwald ngreenwald left a comment

Choose a reason for hiding this comment

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

Looks good! This is the one that had best performance in the benchmarking?

@JLrumberger
Copy link
Collaborator Author

Looks good! This is the one that had best performance in the benchmarking?

Yep

@JLrumberger JLrumberger merged commit f1dae91 into main Feb 6, 2024
5 checks passed
@JLrumberger JLrumberger deleted the training_updates branch February 6, 2024 09:22
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