diff --git a/scripts/train.py b/scripts/train.py index 80d9e953..65db5e37 100644 --- 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()