fix export quant_model

This commit is contained in:
cuicheng01 2021-12-03 07:39:31 +00:00
parent 99b3c96252
commit 998f335b40

View File

@ -373,7 +373,7 @@ class Engine(object):
"inference") "inference")
if self.quanter: if self.quanter:
self.quanter.save_quantized_model( self.quanter.save_quantized_model(
model, model.base_model,
save_path, save_path,
input_spec=[ input_spec=[
paddle.static.InputSpec( paddle.static.InputSpec(