V1版本控制流算子导致内存不足
2022/10/11
105
问题信息
| 问题来源 | 产品大类 | 产品子类 | 关键字 |
|---|---|---|---|
| 官方 | 模型训练 | TensorFlow | 控制流算子、内存不足 |
问题现象描述
模型执行时报错，内存不足。
原因分析
发现该网络的图结构中有switch->merge的V1控制结构，是因为当网络中的分支结构较多，且采用V1版本的控制流算子可能会影响内存复用的效果，导致内存不足。
解决措施
配置环境变量，将V1版本的控制流算子转换为V2版本。
export ENABLE_FORCE_V2_CONTROL=1
本页内容