Enable to use GPU for teklia_dan_predict
https://gitlab.com/teklia/atr/dan/-/blob/main/dan/predict/prediction.py#L420
replace to
device = "cuda:1" if torch.cuda.is_available() else "cpu"
https://gitlab.com/teklia/atr/dan/-/blob/main/dan/predict/prediction.py#L420
replace to
device = "cuda:1" if torch.cuda.is_available() else "cpu"
changed milestone to %ML Prod - Next
assigned to @schneider-y
By M Generali on 2023-07-05T10:50:35 (imported from GitLab)
assigned to @manonBlanco and unassigned @schneider-y
mentioned in merge request !192 (merged)
closed