Преобразовать график обучения в граф вывода?(Удалить нормализацию партии в TF)

Question

Я пытался найти инструменты, которые можно использовать для удаления узлов нормализации партии на этапе вывода.

Я использовал инструмент преобразования графиков, который предоставляет TF, но он фактически не удаляет слои пакетной нормы.Просто соединяет его с предыдущими слоями конвона.

Есть ли у кого-нибудь инструменты, которые они используют для этой цели?

Я бы хорошо с пакетными слоями, но жизнь пока не поддерживает.

MobileNet, кажется, требует пакетной нормализации, так как tflite может запускать MobileNet?

Answer 1

Когда вы конвертируете свой график тензорного потока в граф tflite, утилита tflite_convert (или toco) автоматически объединяет пакетную норму во время разговора.Это часть общих преобразований преобразования