Update worker_training.py

C-Achard · C-Achard · commit da1f14938523 · 2023-07-19T12:01:49.000+02:00
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -643,6 +643,7 @@ def get_loader_func(num_samples):
                             )  #
                             post_label = EnsureType()
 
+                            output_raw = [t for t in pred]
                             val_outputs = [
                                 post_pred(res_tensor) for res_tensor in pred
                             ]
@@ -667,7 +668,7 @@ def get_loader_func(num_samples):
 
                         checkpoint_output.append(
                             [
-                                val_outputs[0].detach().cpu(),
+                                output_raw[0].detach().cpu(),
                                 val_inputs[0].detach().cpu(),
                                 val_labels[0].detach().cpu(),
                             ]

Original file line number	Diff line number	Diff line change
`@@ -643,6 +643,7 @@ def get_loader_func(num_samples):`
`643`	`643`	`) #`
`644`	`644`	`post_label = EnsureType()`
`645`	`645`
	`646`	`+ output_raw = [t for t in pred]`
`646`	`647`	`val_outputs = [`
`647`	`648`	`post_pred(res_tensor) for res_tensor in pred`
`648`	`649`	`]`
`@@ -667,7 +668,7 @@ def get_loader_func(num_samples):`
`667`	`668`
`668`	`669`	`checkpoint_output.append(`
`669`	`670`	`[`
`670`		`- val_outputs[0].detach().cpu(),`
	`671`	`+ output_raw[0].detach().cpu(),`
`671`	`672`	`val_inputs[0].detach().cpu(),`
`672`	`673`	`val_labels[0].detach().cpu(),`
`673`	`674`	`]`