diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index bb35d4f05dc5efc5eb427c7c74e73a9af2251989..4242bb656f4356b42afd00a25033efcb03687b30 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -10,8 +10,8 @@ Loading: stage: build script: - echo "dataset testing" - - cd /data_era5 - - ls -ls +# - cd /data_era5 +# - ls -ls EnvSetup: diff --git a/video_prediction_tools/main_scripts/main_visualize_postprocess.py b/video_prediction_tools/main_scripts/main_visualize_postprocess.py index dd26a2c4c956aee0b8a42f4334732e0b0c53f028..deab9c03666f56237c937622b2613f16fb7be249 100644 --- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py +++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py @@ -1268,6 +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)") args = parser.parse_args() method = os.path.basename(__file__) @@ -1293,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) + 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()