未验证 提交 3801f7aa 编写于 作者: G guofei 提交者: GitHub

Fix the typo in export_model (#2319)

上级 bad37a89
...@@ -386,7 +386,7 @@ class Trainer(object): ...@@ -386,7 +386,7 @@ class Trainer(object):
}] }]
# dy2st and save model # dy2st and save model
if self.slim is None or self.cfg['slim'] != 'quant': if self.slim is None or self.cfg['slim'] != 'QAT':
static_model = paddle.jit.to_static( static_model = paddle.jit.to_static(
self.model, input_spec=input_spec) self.model, input_spec=input_spec)
# NOTE: dy2st do not pruned program, but jit.save will prune program # NOTE: dy2st do not pruned program, but jit.save will prune program
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册