Fix merge issue in image_logger

SimonLarsen · Jan 15, 2024 · 4a50a60 · 4a50a60
1 parent 325ecf8
commit 4a50a60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stort/callbacks/image_logger.py b/stort/callbacks/image_logger.py
@@ -87,7 +87,7 @@ def _callback(pipeline: SupervisedPipeline):
             )
 
         images = []
-        for batch in pbar:
+        for batch in data_iter:
             x, y = prepare_batch(batch, device, non_blocking=False)
             x = input_transform(x)
             with torch.inference_mode():