diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py
index dc9c62a887cd129b1e05d07acc4e2f585c685e60..30f959806549d876a44e3f894f6ccc56508714cb 100644
--- a/modules/deeplearning/srcnn.py
+++ b/modules/deeplearning/srcnn.py
@@ -636,7 +636,7 @@ class SRCNN:
 
         self.reset_test_metrics()
 
-        pred = self.model([data])
+        pred = self.model([data], training=False)
         self.test_probs = pred
         pred = pred.numpy()