将训练图转换为推理图? (删除TF中的批量标准化)

时间:2019-03-01 01:31:59

标签: tensorflow tensorflow-lite batch-normalization

我一直在尝试寻找可用于在推理阶段删除批处理规范化节点的工具。

我使用了TF提供的图形转换工具,但实际上并没有删除批处理规范层。只需将其融合到先前的转换层即可。

是否有人为此目的使用任何工具?

我对批处理层没问题,但是tf life还不支持它。

MobileNet似乎需要批量标准化,那么tflite如何能够运行MobileNet?

1 个答案:

答案 0 :(得分:0)

将张量流图转换为tflite图时,实用程序tflite_convert(或toco)会在对话期间自动融合批处理规范。这是转换常规转换的一部分