提交 99298c8a 编写于 作者: W weishengyu

change default config

上级 91f72dce
...@@ -38,7 +38,7 @@ class ExportModel(nn.Layer): ...@@ -38,7 +38,7 @@ class ExportModel(nn.Layer):
super().__init__() super().__init__()
self.base_model = build_model(config) self.base_model = build_model(config)
self.infer_output_key = config.get("infer_output_key") 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) self.softmax = nn.Softmax(axis=-1)
else: else:
self.softmax = None self.softmax = None
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册