Allow specifying device to use when training

Closes #139 (closed)

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