fix slim export model

This commit is contained in:
LDOUBLEV 2022-04-11 05:13:56 +00:00
parent d76bc57ab9
commit 1c589d6a3a

View File

@ -133,6 +133,7 @@ def main():
sub_model_save_path, logger)
else:
save_path = os.path.join(save_path, "inference")
model.eval()
export_single_model(quanter, model, infer_shape, save_path, logger)