--disable_reuse_memory
功能说明
内存复用开关。
内存复用是指按照生命周期和内存大小,把不冲突的内存重复使用,来降低网络内存占用。
关联参数
无。
参数取值
- 1:关闭内存复用。如果网络模型较大,关闭内存复用开关,模型转换时可能会造成内存不足,导致模型编译失败。
- 0:开启内存复用。默认为0。
推荐配置及收益
无。
示例
--disable_reuse_memory=0
支持的型号
Atlas 200/300/500 推理产品
Atlas 推理系列产品
Atlas 训练系列产品
Atlas 200/500 A2推理产品
Atlas A2训练系列产品/Atlas 800I A2推理产品
依赖约束
在内存复用场景下(默认开启内存复用),支持基于指定算子(节点名称/算子类型)单独分配内存。通过OP_NO_REUSE_MEM环境变量指定要单独分配的一个或多个节点,支持混合配置。配置多个节点时,中间通过英文逗号(“,”)隔开。详细说明请参见《环境变量参考》。
- 基于节点名称配置
节点名称需要配置为转换为CANN平台网络后的节点名称,此节点名称可从Profiling调优数据中获取,详细可参见《性能分析工具使用指南》。
export OP_NO_REUSE_MEM=gradients/logits/semantic/kernel/Regularizer/l2_regularizer_grad/Mul_1,resnet_v1_50/conv1_1/BatchNorm/AssignMovingAvg2
- 基于算子类型配置
export OP_NO_REUSE_MEM=FusedMulAddN,BatchNorm
- 混合配置
export OP_NO_REUSE_MEM=FusedMulAddN,resnet_v1_50/conv1_1/BatchNorm/AssignMovingAvg
父主题: 模型调优选项