Skip to content

S
seminar-breakout
Commit 42416945 authored by Edgar Liberis's avatar Edgar Liberis Committed by Yuxin Wu
Browse files
Options

Don't add keras phase callback if learning phase is fixed (#1205)

parent e97ce38d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/contrib/keras.py
View file @ 42416945
...@@ -217,7 +217,7 @@ def setup_keras_trainer( ...@@ -217,7 +217,7 @@ def setup_keras_trainer(
input, input,
get_cost, get_cost,
lambda: optimizer) lambda: optimizer)
if len(keras.backend.learning_phase().consumers()) > 0: if isinstance(keras.backend.learning_phase(), tf.Tensor) and len(keras.backend.learning_phase().consumers()) > 0:
# check if learning_phase is used in this model # check if learning_phase is used in this model
trainer.register_callback(KerasPhaseCallback(True)) trainer.register_callback(KerasPhaseCallback(True))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment