change default config

2021-06-05 18:05:08 +08:00 · 2021-06-05 18:05:08 +08:00 · 99298c8a15
parent 91f72dcef7
commit 99298c8a15
1 changed files with 1 additions and 1 deletions
--- a/tools/export_model.py
+++ b/tools/export_model.py
@ -38,7 +38,7 @@ class ExportModel(nn.Layer):
        super().__init__()
        self.base_model = build_model(config)
        self.infer_output_key = config.get("infer_output_key")
-        if config.get("infer_add_softmax", False):
+        if config.get("infer_add_softmax", True):
            self.softmax = nn.Softmax(axis=-1)
        else:
            self.softmax = None