log level info

alan-turing-institute · Jul 25, 2024 · 25e5362 · 25e5362
1 parent 5adce7a
commit 25e5362
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/train.py b/scripts/train.py
@@ -1,4 +1,5 @@
 import argparse
+import logging
 import shutil
 
 import wandb
@@ -24,6 +25,8 @@
 
 
 def main(experiment_path):
+    logging.basicConfig(level=logging.INFO)
+
     # Step 0: get start time
     start_time = get_datetime_str()