micronet icon indicating copy to clipboard operation
micronet copied to clipboard

bn融合训练时就没有bn了吗相当于?

Open ArtyZe opened this issue 3 years ago • 1 comments

您好,我看您代码,理解着是只要开启了bn融合后,训练时就相当于没有bn参数的更新了,就用之前的rolling mean 和variance了吗?

ArtyZe avatar Mar 11 '21 04:03 ArtyZe

qat中,若加载预训练浮点模型参数,bn则在其running参数基础上继续更新;在最近更新的qaft中,会冻结。 bn融合指的是,量化训练时bn参数依然存在,但不做bn运算,而是(qat中)更新后融合到前面的conv中;量化推理时,bn融合后完全消除。 可查看相关代码。

666DZY666 avatar May 12 '21 09:05 666DZY666