From 95f693e92a31e7b4a4de8d308c87dd8e84a3abf5 Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Sun, 14 Feb 2021 15:41:38 +0100
Subject: [PATCH] Correct path-construction to data split-templates.
---
.../config_runscripts/config_training.py | 15 +++++++++------
1 file changed, 9 insertions(+), 6 deletions(-)
diff --git a/video_prediction_tools/config_runscripts/config_training.py b/video_prediction_tools/config_runscripts/config_training.py
index 4363bda1..fcd3c019 100755
--- a/video_prediction_tools/config_runscripts/config_training.py
+++ b/video_prediction_tools/config_runscripts/config_training.py
@@ -104,11 +104,12 @@ class Config_Train(Config_runscript_base):
os.makedirs(self.destination_dir)
# Create json-file for data splitting
- source_datasplit = os.path.join("..", "data_split", "datasplit_template.json")
+ source_datasplit = os.path.join("..", "data_split", self.dataset, "datasplit_template.json")
dest_datasplit = os.path.join(self.destination_dir, "data_split.json")
# sanity check (default data_split json-file exists)
if not os.path.isfile(source_datasplit):
- raise FileNotFoundError("%{0}: Could not find default data_split json-file '{1}'".format(method_name, source_datasplit))
+ raise FileNotFoundError("%{0}: Could not find default data_split json-file '{1}'".format(method_name,
+ source_datasplit))
# ...copy over json-file for data splitting...
os.system("cp "+source_datasplit+" "+dest_datasplit)
# ...and open vim after some delay
@@ -119,16 +120,18 @@ class Config_Train(Config_runscript_base):
sp.call("sed -i '/^#/d' {0}".format(dest_datasplit), shell=True)
# Create json-file for hyperparameters
- source_hparams = os.path.join("..","hparams", self.dataset, self.model, "model_hparams.json")
+ source_hparams = os.path.join("..","hparams", self.dataset, self.model, "model_hparams_template.json")
+ dest_hparams = os.path.join(self.destination_dir, "model_hparams.json")
# sanity check (default hyperparameter json-file exists)
if not os.path.isfile(source_hparams):
- raise FileNotFoundError("%{0}: Could not find default hyperparameter json-file '%{1}'".format(method_name, source_hparams))
+ raise FileNotFoundError("%{0}: Could not find default hyperparameter json-file '%{1}'".format(method_name,
+ source_hparams))
# ...copy over json-file for hyperparamters...
- os.system("cp "+source_hparams+" "+self.destination_dir)
+ os.system("cp "+source_hparams+" "+dest_hparams)
# ...and open vim after some delay
print("*** Please configure the model hyperparameters:")
time.sleep(3)
- cmd_vim = os.environ.get('EDITOR', 'vi') + ' ' + os.path.join(self.destination_dir, "model_hparams.json")
+ cmd_vim = os.environ.get('EDITOR', 'vi') + ' ' + dest_hparams
sp.call(cmd_vim, shell=True)
#
# -----------------------------------------------------------------------------------
--
GitLab