-
Notifications
You must be signed in to change notification settings - Fork 17
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
Add 2D Probabilistic UNet Training Setup #123
Conversation
anwai98
commented
Apr 27, 2023
- PUNet Source Training Workflow
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.
I left a few minor comments; looks good overall.
experiments/probabilistic_domain_adaptation/livecell/punet_source.py
Outdated
Show resolved
Hide resolved
The PUNet Source workflow is completed now. Just updated the pipeline for source evaluations. Thanks! |
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.
One minor comment, otherwise this looks good.
I am looking into the test failure here now: #124
@@ -23,6 +23,7 @@ | |||
from torch_em.util import load_model | |||
|
|||
CELL_TYPES = ["A172", "BT474", "BV2", "Huh7", "MCF7", "SHSY5Y", "SkBr3", "SKOV3"] | |||
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu") |
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.
This is not a good idea to declare globally, and I also don't see that you use it anywhere.