未验证 提交 75a1e1e9 编写于 作者: C ceci3 提交者: GitHub

fix strategy (#1368)

上级 3819255b
......@@ -389,7 +389,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.
先完成此消息的编辑!
想要评论请 注册