upload

main.py

@@ -78,9 +78,6 @@ memory_rate_max={memory_max_utility}
 constrain_total={constrain_total}
 constrain_mine={"true" if constrain_mine else "false"}
 constrain_rate={"true" if constrain_rate else "false"}
-date_stamp=`date '+%Y-%m-%d-%H'`
-log_dir="logs/$exp_name/$date_stamp"
-mkdir -p $log_dir
 """
 
 

parse_code.py

@@ -35,6 +35,9 @@ def parse_base_code(contents, debug = False):
                 """cd $(cd "$(dirname "$0")";pwd); source gpu_utility.sh\n\n"""
                 , "device_control"))
             spans_with_type_added_device_control.append((span, type_))
+            spans_with_type_added_device_control.append(("""date_stamp=`date '+%Y-%m-%d-%H'`
+                                                         log_dir="logs/$exp_name/$date_stamp"
+                                                         mkdir -p $log_dir\n""", "logging"))
             # spans_with_type_added_device_control.append((gpu_env, "device_control"))
             # spans_with_type_added_device_control.append((update_device_func, "device_control"))
         elif type_ == "loop":