--external_weight
功能说明
生成om模型文件时,是否将原始网络中的Const/Constant节点的权重保存在单独的文件中,同时将节点类型转换为FileConstant类型。
适用于对om模型文件大小有限制的场景。
关联参数
需要和--output参数配合使用,生成的权重文件保存在与om文件同层级的weight目录下,权重文件以算子名称命名。
参数取值
参数值:
- 0:不将原始网络中的Const/Constant节点的权重保存在单独的文件中,而是直接保存在om模型文件中。
- 1:将原始网络中的Const/Constant节点的权重保存在单独的文件中,且该文件保存在与om文件同级的weight目录下,权重文件以算子名称命名。
参数默认值:0
推荐配置及收益
对于对om模型文件大小有限制的场景,推荐配置为1。
示例
以ONNX网络模型为例:
atc --framework=5 --model=$HOME/module/resnet50.onnx --output=$HOME/module/out/onnx_resnet50 --soc_version=<soc_version> --external_weight=1
支持的芯片型号
Atlas 200/500 A2推理产品
依赖约束
在模型转换时,若将--external_weight参数设置为1,则在使用AscendCL接口开发推理应用、加载模型时,仅支持从om模型文件加载模型,同时需将weight目录与om文件放在同级目录下,否则AscendCL无法加载weight目录下的权重文件。
父主题: 功能配置选项