Skip to content
Snippets Groups Projects
Commit 77032fab authored by Bing Gong's avatar Bing Gong
Browse files

Update main_visualize_postprocess.py

parent e6fea726
Branches
Tags
1 merge request!6CICD testing for postprocessing
Pipeline #93336 passed
......@@ -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()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment