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
......@@ -101,7 +101,10 @@ class GenericTrainingManager:
def init_hardware_config(self):
# 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_amp"] = False
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment