Update worker_training.py

Author: C-Achard

napari_cellseg3d/code_models/worker_training.py

@@ -530,7 +530,7 @@ def get_loader_func(num_samples):
 
             self.log_parameters()
 
-            device = self.config.device
+            device = torch.device(self.config.device)
 
             # if model_name == "test":
             #     self.quit()