diff --git a/video_prediction_tools/main_scripts/main_visualize_postprocess.py b/video_prediction_tools/main_scripts/main_visualize_postprocess.py index 6d763f4de9fe605a54e4db0bb78ef4ca523ef96f..deab9c03666f56237c937622b2613f16fb7be249 100644 --- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py +++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py @@ -1268,7 +1268,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)") + 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__) @@ -1294,7 +1295,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,frac_data=args.frac_data) + 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()