Add infer batch counter

golmschenk · Jul 27, 2024 · f5f2b72 · f5f2b72
1 parent e97dd94
commit f5f2b72
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/qusi/internal/infer_session.py b/src/qusi/internal/infer_session.py
@@ -46,5 +46,6 @@ def infer_phase(dataloader, model: Module, device: Device):
             batch_predicted_targets = model(input_features_on_device)
             batches_of_predicted_targets.append(batch_predicted_targets.cpu().numpy())
             batch_count += 1
+            print(f'Batches completed: {batch_count}')
     predicted_targets = np.concatenate(batches_of_predicted_targets, axis=0)
     return predicted_targets