fixes eval bug

HuiZeng · Mar 31, 2018 · 2d30dc8 · 2d30dc8
1 parent 5ac95ef
commit 2d30dc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()