Skip to content

Commit f5f2b72

Browse files
committed
Add infer batch counter
1 parent e97dd94 commit f5f2b72

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/qusi/internal/infer_session.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,5 +46,6 @@ def infer_phase(dataloader, model: Module, device: Device):
4646
batch_predicted_targets = model(input_features_on_device)
4747
batches_of_predicted_targets.append(batch_predicted_targets.cpu().numpy())
4848
batch_count += 1
49+
print(f'Batches completed: {batch_count}')
4950
predicted_targets = np.concatenate(batches_of_predicted_targets, axis=0)
5051
return predicted_targets

0 commit comments

Comments
 (0)