提交 2ab10a2d 编写于 作者: S stephon

Merge branch 'binaryindex_build_search' of...

Merge branch 'binaryindex_build_search' of https://github.com/Intsigstephon/PaddleClas into binaryindex_build_search
......@@ -6,4 +6,4 @@ class IdentityHead(nn.Layer):
super(IdentityHead, self).__init__()
def forward(self, x, label=None):
return {"features": x, "logits": None}
\ No newline at end of file
return {"features": x, "logits": None}
# global configs
Global:
checkpoints: null
pretrained_model: null
pretrained_model: ./output_product_binary/best_model
output_dir: ./output_dlbhc/
device: gpu
save_interval: 1
......@@ -34,6 +34,7 @@ Arch:
infer_output_key: "features"
infer_add_softmax: "false"
infer_binarize: "round"
# loss function config for train/eval process
Loss:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册