Skip to content

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"