Add logging level (#237)

SainsburyWellcomeCentre · Nov 5, 2024 · 420e0f7 · 420e0f7
1 parent b72d4ad
commit 420e0f7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/crabs/tracker/track_video.py b/crabs/tracker/track_video.py
@@ -336,6 +336,8 @@ def tracking_parse_args(args):
 
 def app_wrapper():
     """Wrap function to run the tracking application."""
+    logging.getLogger().setLevel(logging.INFO)
+
     torch.set_float32_matmul_precision("medium")
 
     tracking_args = tracking_parse_args(sys.argv[1:])