下载
中文
注册
校准过程中提示"IFMR node. Name:'layer_ifmr_op' Status Message: std::bad_alloc"信息

校准过程中提示"IFMR node. Name:'layer_ifmr_op' Status Message: std::bad_alloc"信息

2025/04/17

23

暂无评分
我要评分

问题信息

问题来源产品大类产品子类关键字
官方应用开发CANNIFMR 、layer_ifmr_op、bad_alloc

问题现象描述

训练后量化校准过程中,对某层进行量化,提示内存错误"IFMR node. Name:'layer_ifmr_op' Status Message: std::bad_alloc "。

原因分析

相较于环境中的CPU或者GPU存储,量化时使用的数据量过大,导致内存错误。

解决措施

  1. 减少校准所用的数据量,即降低batch_size或者batch_num,可直接使用batch_size * batch_num的最小值尝试量化。
  2. 如果网络中仅个别层的featuremap很大,出现内存错误,则可以跳过该层,不对其进行量化。

本页内容