search

vanvalenlab / cellSAM

Codebase for "A Foundation Model for Cell Segmentation"
Other
51 stars 7 forks source link

Make model loadable on non-cuda systems #15

Open rossbar opened 6 months ago

rossbar commented 6 months ago

Maps the model to CPU by default, so that the model can be loaded on systems without CUDA support.