未验证 提交 a523f6b3 编写于 作者: G Guanghua Yu 提交者: GitHub

fix QAT export bug (#52218)

上级 f86d0be7
......@@ -530,6 +530,7 @@ class ImperativeQuantizeOutputs:
model, paddle.nn.Layer
), "The model must be the instance of paddle.nn.Layer."
paddle.jit.to_static(model, input_spec=input_spec)
paddle.jit.save(layer=model, path=path, input_spec=input_spec, **config)
is_dynamic_mode = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册