1. 首页
  2. 数据库
  3. 其它
  4. keras_to_tensorflow:将经过训练的keras模型转换为推断张量流模型的通用代码 源码

keras_to_tensorflow:将经过训练的keras模型转换为推断张量流模型的通用代码 源码

上传者: 2021-02-19 00:03:20上传 ZIP文件 12.17KB 热度 21次
Keras到TensorFlow keras_to_tensorflow是将经过训练的keras模型转换为可随时推断的TensorFlow模型的工具。 该工具不是为TensorFlow 2.0量身定制的,它看起来运行良好。 概要 按照默认行为,此工具将冻结节点(将所有TF变量转换为TF常量),并将推理图和权重保存到二进制protobuf(.pb)文件中。 在冻结期间,TensorFlow还应用节点修剪,该修剪将删除对输出张量无贡献的节点。 该工具支持多个输出网络,并使用户能够通过--output_nodes_prefix标志重命名输出张量。 如果设置了--output_meta_ckpt
用户评论