Skip to content
Snippets Groups Projects
Commit 09c6fb94 authored by Manon Blanco's avatar Manon Blanco Committed by Manon Blanco
Browse files

Fix lint

parent c1ca208f
No related branches found
No related tags found
1 merge request!299Allow specifying device to use when training
This commit is part of merge request !299. Comments created here will be created in the context of that merge request.
Hide whitespace changes
Inline Side-by-side
dan/ocr/manager/training.py
+ 4
− 1
View file @ 09c6fb94
...@@ -101,7 +101,10 @@ class GenericTrainingManager: ...@@ -101,7 +101,10 @@ class GenericTrainingManager:
def init_hardware_config(self): def init_hardware_config(self):
# Debug mode # Debug mode
if self.device_params["force"] not in [None, "cuda"] or not torch.cuda.is_available(): if (
self.device_params["force"] not in [None, "cuda"]
or not torch.cuda.is_available()
):
self.device_params["use_ddp"] = False self.device_params["use_ddp"] = False
self.device_params["use_amp"] = False self.device_params["use_amp"] = False
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment