diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 06fbb3107909e3c170bd621aef86c91b126dfd5d..7e91a8329c68f307f6726e26481b6e7f3cb1e3d0 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -740,7 +740,7 @@ class ESPCN:
 
         self.test_preds = preds
 
-    def do_evaluate(self, nda_lr, ckpt_dir, prob_thresh=0.5):
+    def do_evaluate(self, nda_lr, ckpt_dir):
 
         ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
         ckpt_manager = tf.train.CheckpointManager(ckpt, ckpt_dir, max_to_keep=3)