From 09c6fb941e776f0baaa94e2068771c6fc0a060b6 Mon Sep 17 00:00:00 2001
From: manonBlanco <blanco@teklia.com>
Date: Wed, 25 Oct 2023 14:35:01 +0200
Subject: [PATCH] Fix lint
---
dan/ocr/manager/training.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/dan/ocr/manager/training.py b/dan/ocr/manager/training.py
index 28183848..82cfa763 100644
--- a/dan/ocr/manager/training.py
+++ b/dan/ocr/manager/training.py
@@ -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
--
GitLab