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"