From c1c27e987a861ca83801cb09537de4ece7e9198c Mon Sep 17 00:00:00 2001 From: C-Achard Date: Thu, 14 Dec 2023 15:17:22 +0100 Subject: [PATCH] Remove num_classes in worker_inference --- napari_cellseg3d/code_models/worker_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/napari_cellseg3d/code_models/worker_inference.py b/napari_cellseg3d/code_models/worker_inference.py index 7936363b..c69dfe45 100644 --- 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)