内存复用开关。
无。
参数值:
参数默认值:0
参数值约束:如果网络模型较大,关闭内存复用开关,模型转换时可能会造成内存不足,导致模型转换失败。
无。
--disable_reuse_memory=0
昇腾310 AI处理器
昇腾310P AI处理器
昇腾910 AI处理器
在内存复用场景下(默认开启内存复用),支持基于指定算子(节点名称/算子类型)单独分配内存。通过OP_NO_REUSE_MEM环境变量指定要单独分配的一个或多个节点,支持混合配置。配置多个节点时,中间通过英文逗号(“,”)隔开。
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