search

DevoLearn / devolearn

Accelerate data driven research on embryos with deep learning models :microscope:
MIT License
43 stars 30 forks source link

Partial GPU support #4

Closed Mayukhdeb closed 3 years ago

Mayukhdeb commented 4 years ago

Only the lineage_population_model runs on a GPU optionally, need to add CUDA support on all models so that the users can use with models with:

model = example_model_class(device = "cuda")

Mainakdeb commented 4 years ago

I'll work on this