Remove num_classes in worker_inference

C-Achard · C-Achard · commit c1c27e987a86 · 2023-12-14T15:17:22.000+01:00
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)

Original file line number	Diff line number	Diff line change
`@@ -859,7 +859,7 @@ def inference(self):`
`859`	`859`	`model = model_class(`
`860`	`860`	`input_img_size=[dims, dims, dims],`
`861`	`861`	`# device=self.config.device,`
`862`		`- num_classes=self.config.model_info.num_classes,`
	`862`	`+ # num_classes=self.config.model_info.num_classes,`
`863`	`863`	`)`
`864`	`864`	`try:`
`865`	`865`	`model = model.to(self.config.device)`