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

Cpu-based inference #31

Merged
merged 3 commits into from
Feb 1, 2021
Merged

Cpu-based inference #31

merged 3 commits into from
Feb 1, 2021

Conversation

akhanf
Copy link
Member

@akhanf akhanf commented Feb 1, 2021

Thanks to @ylugithub there is a nnunet fork that works on CPU or GPU for inference. This PR updates the dependencies to use that fork (will replace this with a future release of nnunet once it becomes integrated), and adds a couple cmd-line options for enabling gpu (--use_gpu, which sets the resources for the run_inference rule), and a --nnunet_disable_tta to disable test-time augmentation.

By default use_gpu is set to false, although this really only makes a difference to snakemake profiles that use the gpus resource (e.g. cc-slurm).. maybe could change it to --request_gpu_resources to be clear that it is not enabling/disabling GPU more generally..

@akhanf akhanf merged commit be4a7c4 into master Feb 1, 2021
@akhanf akhanf deleted the cpu_inference branch February 26, 2021 22:30
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.

1 participant