Update worker_training.py

C-Achard · C-Achard · commit f109889e0037 · 2023-07-19T11:07:38.000+02:00
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -666,7 +666,7 @@ def get_loader_func(num_samples):
                                 .astype(np.uint16),
                             ]
                         )
-                        [np.squeeze(vol) for vol in checkpoint_output]
+                        # [np.squeeze(vol) for vol in checkpoint_output]
 
                         metric = dice_metric.aggregate().detach().item()
                         dice_metric.reset()

Original file line number	Diff line number	Diff line change
`@@ -666,7 +666,7 @@ def get_loader_func(num_samples):`
`666`	`666`	`.astype(np.uint16),`
`667`	`667`	`]`
`668`	`668`	`)`
`669`		`- [np.squeeze(vol) for vol in checkpoint_output]`
	`669`	`+ # [np.squeeze(vol) for vol in checkpoint_output]`
`670`	`670`
`671`	`671`	`metric = dice_metric.aggregate().detach().item()`
`672`	`672`	`dice_metric.reset()`