Feaure: Added Apple silicon support in nb finetuning.#1161
Conversation
Checks if Cuda or MPS (Apple Silicon) is available. Otherwise defaults to GPU.
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
|
Hi @HendrikLamar, Thanks for the PR. This definitely looks good (we missed it right from the very beginning actually hehe). I'll go ahead and merge this once the tests pass! PS. FYI, we have a function under |
|
The tests are broken because the |
|
Things aren't upstream, that's why they are breaking rn. The PR doesn't cause this explicitly. Therefore, I'll go ahead and merge this. Thanks again @HendrikLamar for the PR! |
f463687
into
computational-cell-analytics:dev
* Fix issue in batched inference * Feaure: Added Apple silicon support in nb finetuning. (#1161) Checks if Cuda or MPS (Apple Silicon) is available. Otherwise defaults to GPU. * Simply device allocation in finetuning notebook (#1163) * Simply device allocation in finetuning notebook * Update finetuning python scripts --------- Co-authored-by: Constantin Pape <constantin.pape@informatik.uni-goettingen.de> Co-authored-by: Hendrik Windel <15370203+HendrikLamar@users.noreply.github.com>
Checks if Cuda or MPS (Apple Silicon) is available. Otherwise defaults to GPU.
Checks if Cuda or MPS (Apple Silicon) is available. Otherwise defaults to CPU.