-
Notifications
You must be signed in to change notification settings - Fork 1
Parallelize prediction #22
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
Conversation
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.
Most of this looks good! The only major comment is that I don't see where you start the slurm array job in case of prediction_instances > 1.
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.
Looks good overall! Just a few cosmetic changes. The test error should be resolved when the new torch_em release is pulled (will be on conda-forge soon, see conda-forge/torch_em-feedstock#20 )
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.
Looks good now, and the test passes. I will go ahead and merge it.
Parallelization of the prediction and segmentation using the distance U-Net.
Includes scripts for resizing wrongly scaled cochleas and converting tif to n5.
WIP