diff --git a/runexp.py b/runexp.py index 8bada3a..421dc70 100644 --- a/runexp.py +++ b/runexp.py @@ -299,7 +299,7 @@ def main(memo, traffic_file_list,roadnet_file,num_row,num_col,path_to_data,worke "PATH_TO_DATA": path_to_data, "PATH_TO_PRETRAIN_MODEL": os.path.join("model", "initial", traffic_file), "PATH_TO_PRETRAIN_WORK_DIRECTORY": os.path.join("records", "initial", traffic_file), - "PATH_TO_ERROR": os.path.join("errors", memo) + "PATH_TO_ERROR": os.path.join("errors", memo), "PATH_TO_PRETRAIN_DATA": "data/template", "PATH_TO_AGGREGATE_SAMPLES": "records/initial" } @@ -342,4 +342,4 @@ def main(memo, traffic_file_list,roadnet_file,num_row,num_col,path_to_data,worke #memo = "multi_phase/optimal_search_new/new_headway_anon" os.environ["CUDA_VISIBLE_DEVICES"] = args.visible_gpu - main(args.memo, ['hangzhou.json'],'roadnet_4_4.json',4,4,'data/Hangzhou',args.workers) \ No newline at end of file + main(args.memo, ['hangzhou.json'],'roadnet_4_4.json',4,4,'data/Hangzhou',args.workers)