Add set_learning_phase in TF backend.

chardmeier · Apr 14, 2016 · 2cc9ebf · 2cc9ebf
1 parent cb5d69c
commit 2cc9ebf
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/keras/backend/tensorflow_backend.py b/keras/backend/tensorflow_backend.py
@@ -21,6 +21,11 @@ def learning_phase():
     return _LEARNING_PHASE
 
 
+def set_learning_phase(value):
+    global _LEARNING_PHASE
+    _LEARNING_PHASE = tf.constant(value, name='keras_learning_phase')
+
+
 def get_session():
     '''Returns the TF session in use by the backend.
     '''