未验证 提交 f9b2081c 编写于 作者: u010070587's avatar u010070587 提交者: GitHub

Merge pull request #891 from FredHuang16/patch-14

add infer_output_key to config
......@@ -19,6 +19,8 @@ Global:
# model architecture
Arch:
name: "RecModel"
infer_output_key: "features"
infer_add_softmax: False
Backbone:
name: "ResNet50_last_stage_stride1"
pretrained: True
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册