文件名称:tfhub-exporter:导出和优化Tensorflow Hub模型以进行推理和部署
文件大小:11KB
文件格式:ZIP
更新时间:2024-02-25 23:33:24
deployment tool exporter tensorflow optimization
Tensorflow集线器模块导出器 该工具允许您创建上可用的预训练机器学习模块的优化版本。 它从TF Hub下载模块,冻结网络权重,应用(可配置)转换次数,以优化其部署和推断能力,并将处理后的模型导出为 。 先决条件 确保您至少运行python 3.6 运行pip install -r requirements.txt以安装必要的依赖项 就是这样,您应该准备就绪。 用法 该存储库中的示例使用模型的。 查看,以获取有关如何导出和加载优化模型的完整示例。 导出TF集线器模块(导出) 用法 python main.py export --help Usage: main.py export
【文件预览】:
tfhub-exporter-master
----usage.ipynb(12KB)
----main.py(5KB)
----LICENSE(1KB)
----requirements.txt(104B)
----.gitignore(1KB)
----lib()
--------tf_graph_utils.py(3KB)
--------cli_utils.py(1KB)
--------__init__.py(362B)
----README.md(5KB)