From 509b864d06771051627afe6c382643df202d2085 Mon Sep 17 00:00:00 2001
From: Yan Ji <y.ji@fz-juelich.de>
Date: Sat, 12 Jun 2021 13:47:18 +0200
Subject: [PATCH] add acc and ssim to boostrapping and .nc file
---
.../main_scripts/main_visualize_postprocess.py | 2 +-
1 file changed, 1 insertion(+), 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 b914c7f9..0c650411 100644
--- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py
+++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
@@ -1229,7 +1229,7 @@ def main():
parser.add_argument("--num_stochastic_samples", type=int, default=1)
parser.add_argument("--gpu_mem_frac", type=float, default=0.95, help="fraction of gpu memory to use")
parser.add_argument("--seed", type=int, default=7)
- parser.add_argument("--evaluation_metrics", "-eval_metrics", dest="eval_metrics", nargs="+", default=("mse", "psnr"),
+ parser.add_argument("--evaluation_metrics", "-eval_metrics", dest="eval_metrics", nargs="+", default=("mse", "psnr", "ssim","acc"),
help="Metrics to be evaluate the trained model. Must be known metrics, see Scores-class.")
parser.add_argument("--channel", "-channel", dest="channel", type=int, default=0,
help="Channel which is used for evaluation.")
--
GitLab