Update train.py

MSiam · Nov 14, 2018 · a5ffffb · a5ffffb · zjj321 · May 9, 2020
1 parent 52f2685
commit a5ffffb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train/train.py b/train/train.py
@@ -911,12 +911,12 @@ def test_inference(self):
             if self.args.random_cropping:
                 feed_dict = {self.test_model.x_pl_before: x_batch,
                              self.test_model.y_pl_before: y_batch
-                             #                             self.test_model.is_training: False,
+                             self.test_model.is_training: False,
                              }
             else:
                 feed_dict = {self.test_model.x_pl: x_batch,
                              self.test_model.y_pl: y_batch
-                             #                             self.test_model.is_training: False
+                             self.test_model.is_training: False
                              }
 
             # calculate the time of one inference