From ac1630491b2917b5fc6217099eb8d7c2c3857b42 Mon Sep 17 00:00:00 2001
From: Bing Gong <b.gong@fz-juelich.de>
Date: Fri, 25 Feb 2022 15:10:38 +0100
Subject: [PATCH] Update main_visualize_postprocess.py
---
.../main_scripts/main_visualize_postprocess.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/video_prediction_tools/main_scripts/main_visualize_postprocess.py b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
index 9618eed4..dffbc898 100644
--- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py
+++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
@@ -1265,6 +1265,8 @@ def main():
help="(Only) metric to evaluate when quick evaluation (-lquick) is chosen.")
parser.add_argument("--climatology_file", "-clim_fl", dest="clim_fl", type=str, default=False,
help="The path to the climatology_t2m_1991-2020.nc file ")
+ parse.add_argument("--frac_data", "-f_dt", dest="f_dt",type=float,default=1,
+ help="fraction of dataset to be used for evaluation (only applied when shuffling is active)")
args = parser.parse_args()
method = os.path.basename(__file__)
@@ -1290,7 +1292,8 @@ def main():
postproc_instance = Postprocess(results_dir=results_dir, checkpoint=args.checkpoint, data_mode="test",
batch_size=args.batch_size, num_stochastic_samples=args.num_stochastic_samples,
gpu_mem_frac=args.gpu_mem_frac, seed=args.seed, args=args,
- eval_metrics=eval_metrics, channel=args.channel, lquick=args.lquick,clim_path=args.clim_fl)
+ eval_metrics=eval_metrics, channel=args.channel, lquick=args.lquick,
+ clim_path=args.clim_fl,frac_data=args.frac_data)
# run the postprocessing
postproc_instance.run()
postproc_instance.handle_eval_metrics()
--
GitLab