diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index e72c1f17a9feae1ee28a86cb4ee5885a9589da41..4c54b2b8b0ae0ed45fec7b35399fe48f33ab4cae 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -315,6 +315,7 @@ class SRCNN:
         # -----------------------------------------------------
         label = input_data[:, label_idx, :, :]
         label = label.copy()
+        # label = np.where(np.isnan(label), 0, label)
         # label = smooth_2d(label, sigma=1.0)
         label = label[:, y_128, x_128]