未验证 提交 341ada8a 编写于 作者: C ceci3 提交者: GitHub

fix strategy (#1369)

上级 13c6a4df
......@@ -395,7 +395,7 @@ class AutoCompression:
config.append(merge_config(quant_config, hpo_config))
### case6: quant_config & distill config ==> QAT & Distill
if quant_config is not None and self._distill_config is not None:
if quant_config is not None and self._distill_config is not None and 'ptq_hpo' not in strategy:
only_distillation = False
strategy.append('qat_dis')
config.append(merge_config(quant_config, self._distill_config))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册