Update ResNet50_ReID.yaml

This commit is contained in:
Felix 2021-06-17 16:33:34 +08:00 committed by GitHub
parent dca87e9818
commit 0fa19dca61
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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