Use GPU for training - Githubissues

autonlab / auton-survival

Auton Survival - an open source package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Events

http://autonlab.github.io/auton-survival

MIT License

321 stars 74 forks source link

Closed Jeanselme closed 3 years ago

Jeanselme commented 3 years ago

Check if GPU is available and train model on GPU (could be more elegant)

chiragnagpal commented 3 years ago

We need to add to a GPU flag to the dsm_api.py module as well ...

Jeanselme commented 3 years ago

Updated, should be more user friendly now