Small updates

facebookresearch · Jun 10, 2021 · 39fa0ba · 39fa0ba
1 parent 227127f
commit 39fa0ba
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vissl/utils/benchmark_suite_scheduler.py b/vissl/utils/benchmark_suite_scheduler.py
@@ -8,6 +8,7 @@
 import logging
 import os
 import time
+from datetime import datetime
 from pathlib import Path
 from typing import List
 
@@ -338,7 +339,7 @@ def _evaluate_checkpoint(self, checkpoint_str, benchmarks):
             benchmark["slurm_state"] = job.state
 
             log = f"""
-                Launched Slurm Evaluation job.
+                Launched Slurm Evaluation job. Time: { datetime.now().strftime("%H:%M:%S %z") }
                 job_id: { job.job_id }, num_retries: { benchmark["num_retries"] }
                 evaluation_name: { benchmark["evaluation_name"] }
                 checkpoint_str: { checkpoint_str }
@@ -402,7 +403,7 @@ def _monitor_benchmark_job(self, benchmark):
             checkpoint_str = os.path.split(benchmark["weights_init_params_file"])[-1]
 
             log = f"""
-                Slurm Evaluation job changed states.
+                Slurm Evaluation job changed states. Time: { datetime.now().strftime("%H:%M:%S %z") }
                 job_id: { job.job_id }, num_retries: { benchmark["num_retries"] }
                 evaluation_name: { benchmark["evaluation_name"] },
                 checkpoint_str: { checkpoint_str },
@@ -574,7 +575,6 @@ def _set_initial_benchmark_result(
             f"config.MODEL.WEIGHTS_INIT.PARAMS_FILE='{weights_init_path}'",
             "config.SLURM.USE_SLURM=true",
             f"config.SLURM.LOG_FOLDER='{log_dir}'",
-            f"config.SLURM.LOG_FOLDER='{log_dir}'",
             f"config.CHECKPOINT.DIR='{checkpoint_dir}'",
             f"hydra.run.dir='{ log_dir }'",
         ]: