Remove num_classes in worker_inference

AdaptiveMotorControlLab · Dec 14, 2023 · c1c27e9 · c1c27e9
1 parent d62f550
commit c1c27e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/napari_cellseg3d/code_models/worker_inference.py b/napari_cellseg3d/code_models/worker_inference.py
@@ -859,7 +859,7 @@ def inference(self):
                 model = model_class(
                     input_img_size=[dims, dims, dims],
                     # device=self.config.device,
-                    num_classes=self.config.model_info.num_classes,
+                    # num_classes=self.config.model_info.num_classes,
                 )
                 try:
                     model = model.to(self.config.device)