lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Bug fix to set correct uses_learning_phase flag

Browse Source 
```test_on_batch``` and ```predict_on_batch``` had the wrong ```uses_learning_phase``` flag
This commit is contained in:
Michael Oliver 2016-04-06 16:56:31 -07:00 committed by François Chollet
parent e179198410
commit 63059f6063
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
keras/engine/training.py
View File

@ -1126,7 +1126,7 @@ class Model(Container):
sample_weight=sample_weight,
check_batch_dim=True)
if self.uses_learning_phase:
ins = x + y + sample_weights + [1.]
ins = x + y + sample_weights + [0.]
else:
ins = x + y + sample_weights
outputs = self.test_function(ins)
@ -1140,7 +1140,7 @@ class Model(Container):
x = standardize_input_data(x, self.input_names,
self.internal_input_shapes)
if self.uses_learning_phase:
ins = x + [1.]
ins = x + [0.]
else:
ins = x
outputs = self.predict_function(ins)