improve tqdm bar

clemsgrs · Mar 7, 2024 · b2c743b · b2c743b
1 parent 5de975a
commit b2c743b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dinov2/inference/extract_features.py b/dinov2/inference/extract_features.py
@@ -121,6 +121,7 @@ def main(args):
         data_loader,
         desc="Feature Extraction",
         unit=" img",
+        unit_scale=cfg.inference.batch_size,
         ncols=80,
         position=0,
         leave=True,