下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

--disable_reuse_memory

功能说明

内存复用开关。

关联参数

无。

参数取值

参数值:

  • 1:关闭内存复用。
  • 0:开启内存复用。

参数默认值:0

参数值约束:如果网络模型较大,关闭内存复用开关,调优时可能会造成内存不足,导致调优失败。

推荐配置及收益

无。

示例

--disable_reuse_memory=0

依赖约束

在内存复用场景下(默认开启内存复用),支持基于指定算子(节点名称/算子类型)单独分配内存。通过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
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词