From eb0981885d1553d136ed8b95fc95a966da5c316d Mon Sep 17 00:00:00 2001 From: gong1 <b.gong@fz-juelich.de> Date: Mon, 12 Jul 2021 17:04:30 +0200 Subject: [PATCH] Bug fix for loading data_dict.json --- .../main_scripts/main_visualize_postprocess.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/video_prediction_tools/main_scripts/main_visualize_postprocess.py b/video_prediction_tools/main_scripts/main_visualize_postprocess.py index 9d008cf0..77334642 100644 --- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py +++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py @@ -143,7 +143,7 @@ class Postprocess(TrainModel): model_opt_js = os.path.join(self.checkpoint, "options.json") model_ds_js = os.path.join(self.checkpoint, "dataset_hparams.json") model_hp_js = os.path.join(self.checkpoint, "model_hparams.json") - model_dd_js = os.path.join(self.checkpoint, "data_dict.json") + model_dd_js = os.path.join(self.checkpoint, "data_split.json") if os.path.isfile(model_opt_js): shutil.copy(model_opt_js, os.path.join(self.results_dir, "options_checkpoints.json")) @@ -161,7 +161,7 @@ class Postprocess(TrainModel): raise FileNotFoundError("%{0}: The file {1} does not exist".format(method_name, model_hp_js)) if os.path.isfile(model_dd_js): - shutil.copy(model_dd_js, os.path.join(self.results_dir, "data_dict.json")) + shutil.copy(model_dd_js, os.path.join(self.results_dir, "data_split.json")) else: raise FileNotFoundError("%{0}: The file {1} does not exist".format(method_name, model_dd_js)) @@ -177,13 +177,13 @@ class Postprocess(TrainModel): """ method_name = Postprocess.load_jsons.__name__ - datasplit_dict = os.path.join(self.results_dir, "data_dict.json") + datasplit_dict = os.path.join(self.results_dir, "data_split.json") model_hparams_dict = os.path.join(self.results_dir, "model_hparams.json") checkpoint_opt_dict = os.path.join(self.results_dir, "options_checkpoints.json") # sanity checks on the JSON-files if not os.path.isfile(datasplit_dict): - raise FileNotFoundError("%{0}: The file data_dict.json is missing in {1}".format(method_name, + raise FileNotFoundError("%{0}: The file data_split.json is missing in {1}".format(method_name, self.results_dir)) if not os.path.isfile(model_hparams_dict): -- GitLab