Update worker_training.py

C-Achard · C-Achard · commit 7e09af2a48b7 · 2023-07-19T12:29:30.000+02:00
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -685,7 +685,7 @@ def get_loader_func(num_samples):
                             for channel in checkpoint_output
                             for item in channel
                         ]
-                        checkpoint_output[3] = checkpoint_output[2].astype(
+                        checkpoint_output[3] = checkpoint_output[3].astype(
                             np.uint16
                         )
 

Original file line number	Diff line number	Diff line change
`@@ -685,7 +685,7 @@ def get_loader_func(num_samples):`
`685`	`685`	`for channel in checkpoint_output`
`686`	`686`	`for item in channel`
`687`	`687`	`]`
`688`		`- checkpoint_output[3] = checkpoint_output[2].astype(`
	`688`	`+ checkpoint_output[3] = checkpoint_output[3].astype(`
`689`	`689`	`np.uint16`
`690`	`690`	`)`
`691`	`691`