Можете ли вы использовать обучающий параметр с конструктором tf.keras.Model ()?

Question

Возможно ли получить правильное значение параметра training, передаваемого слою в модели, созданной с помощью конструктора tf.keras.Model() ("функциональный" API), или вам нужно сделать подкласс tf.keras.Model(), чтобы сделать это ?

Мне нужно training для tf.keras.layers.BatchNormalization.

Answer 1

Краткий ответ: не совсем.
Слой Keras имеет параметр trainable, к которому вы можете обращаться и изменять. Но это больше используется для замораживания весов.

Что касается Sequential API, то на сегодняшний день документации мало о том, как BatchNorm ведет себя в режиме вывода. Есть даже некоторые открытые проблемы GitHub здесь и здесь .

tf.keras.Model() API дает вам больше контроля над потоком и тем, как слои ведут себя, так что если вы не обеспокоенный форматом Saved Model, который я бы придерживался.