From f5f2b7283943358e681a7a6c8d9bedc4ddfacd98 Mon Sep 17 00:00:00 2001 From: golmschenk Date: Fri, 26 Jul 2024 20:48:36 -0400 Subject: [PATCH] Add infer batch counter --- src/qusi/internal/infer_session.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/qusi/internal/infer_session.py b/src/qusi/internal/infer_session.py index f1f1ad4..fd9dc73 100644 --- 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