Как установить опцию пакетной нормализации в режиме вывода без вызова tf.layers.batch_normalization()?
Я определяю глубокий CNN с тензорным потоком, включая опцию пакетной нормализации, т.е. мой код может выглядеть так:
def network(input):
...
input = tf.layers.batch_normalization(input, ...)
...
Предположим, что сеть была обучена, и файл контрольных точек был сохранен. Теперь я хотел бы использовать эту модель для вывода. Обычно я могу вызвать функцию network(input) опять же, за исключением передачи параметра training=False в tf.layers.batch_normalization(), а затем восстановить веса из файла контрольных точек.
Тем не менее, я бы предпочел использовать tf.import_meta_graph перестроить мою сеть, так как код в функции network(input) может быть изменено.
Но теперь, как я могу установить опцию пакетной нормализации в режиме вывода на этот раз? Так как у меня нет доступа к функции tf.layers.batch_normalization()Мне немного сложно это решить.