--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平台网络后的节点名称，节点名称可以通过设置DUMP_GE_GRAPH环境变量，在导出的ge_onnx_xxx_Build.pbtxt最终图中查看“name”字段获取。

  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