freeze_graph.py将变量转换为常量,并删除未在推理中使用的训练节点,也可以应用几个优化。来自已保存模型的graph_def仍包含训练节点和变量。如果你想在saved_model,graphdef上调用TRT优化,你需要先冻结它。您可以在脚本中调用freeze_graph()。检查freeze_graph.py。
float_ref
最新推荐文章于 2023-03-14 15:44:56 发布
freeze_graph.py将变量转换为常量,并删除未在推理中使用的训练节点,也可以应用几个优化。来自已保存模型的graph_def仍包含训练节点和变量。如果你想在saved_model,graphdef上调用TRT优化,你需要先冻结它。您可以在脚本中调用freeze_graph()。检查freeze_graph.py。