未验证 提交 ea53ad95 编写于 作者: B Bin Lu 提交者: GitHub

Update ResNet50.yaml

上级 6391c90b
...@@ -16,7 +16,21 @@ Global: ...@@ -16,7 +16,21 @@ Global:
# model architecture # model architecture
Arch: Arch:
#for classification
name: "ResNet50" name: "ResNet50"
#for retrieval
# name: "RecModel"
# Backbone:
# name: "ResNet50"
# Stoplayer:
# name: "flatten_0"
# output_dim: 2048
# embedding_size: 512
# Head:
# name: "ArcMargin"
# margin: 0.5
# scale: 80
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册