华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
V1版本控制流算子导致内存不足
2022/10/11
446
问题信息
模型执行时报错,内存不足。
发现该网络的图结构中有switch->merge的V1控制结构,是因为当网络中的分支结构较多,且采用V1版本的控制流算子可能会影响内存复用的效果,导致内存不足。
配置环境变量,将V1版本的控制流算子转换为V2版本。
export ENABLE_FORCE_V2_CONTROL=1
本页内容