diff --git a/train/train.py b/train/train.py index 887700c..8f5c086 100644 --- a/train/train.py +++ b/train/train.py @@ -910,12 +910,12 @@ def test_inference(self): # Feed this variables to the network 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.y_pl_before: y_batch, + 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.y_pl: y_batch, self.test_model.is_training: False }