From 2d30dc87641f1764e8f6ec3f917f788c6ad7cc1a Mon Sep 17 00:00:00 2001 From: Michael Gharbi Date: Sat, 31 Mar 2018 11:07:15 -0400 Subject: [PATCH] fixes eval bug --- hdrnet/bin/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hdrnet/bin/train.py b/hdrnet/bin/train.py index 1ae88fd..031fe31 100755 --- a/hdrnet/bin/train.py +++ b/hdrnet/bin/train.py @@ -102,7 +102,7 @@ def main(args, model_params, data_params): eval_prediction = mdl.inference( eval_samples['lowres_input'], eval_samples['image_input'], model_params, is_training=False) - eval_psnr = metrics.psnr(eval_samples['image_output'], prediction) + eval_psnr = metrics.psnr(eval_samples['image_output'], eval_prediction) # Optimizer global_step = tf.contrib.framework.get_or_create_global_step()