提交 cabc7a0f 编写于 作者: G gaotingquan 提交者: Tingquan Gao

fix: replace reg exp with layer name

上级 71cb728b
...@@ -22,7 +22,7 @@ Arch: ...@@ -22,7 +22,7 @@ Arch:
name: "ResNet50" name: "ResNet50"
pretrained: True pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: "flatten_0" name: "flatten"
output_dim: 2048 output_dim: 2048
Head: Head:
name: "FC" name: "FC"
......
...@@ -28,7 +28,7 @@ Arch: ...@@ -28,7 +28,7 @@ Arch:
pretrained: True pretrained: True
use_ssld: True use_ssld: True
BackboneStopLayer: BackboneStopLayer:
name: flatten_0 name: "flatten"
Neck: Neck:
name: FC name: FC
embedding_size: 1280 embedding_size: 1280
......
...@@ -24,7 +24,7 @@ Arch: ...@@ -24,7 +24,7 @@ Arch:
name: "ResNet50_last_stage_stride1" name: "ResNet50_last_stage_stride1"
pretrained: True pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: "adaptive_avg_pool2d_0" name: "avg_pool"
Neck: Neck:
name: "VehicleNeck" name: "VehicleNeck"
in_channels: 2048 in_channels: 2048
......
...@@ -25,7 +25,7 @@ Arch: ...@@ -25,7 +25,7 @@ Arch:
name: ResNet50_vd name: ResNet50_vd
pretrained: True pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: flatten_0 name: "flatten"
Neck: Neck:
name: FC name: FC
embedding_size: 2048 embedding_size: 2048
......
...@@ -25,7 +25,7 @@ Arch: ...@@ -25,7 +25,7 @@ Arch:
name: ResNet50_vd name: ResNet50_vd
pretrained: False pretrained: False
BackboneStopLayer: BackboneStopLayer:
name: flatten_0 name: "flatten"
Neck: Neck:
name: FC name: FC
embedding_size: 2048 embedding_size: 2048
......
...@@ -22,7 +22,7 @@ Arch: ...@@ -22,7 +22,7 @@ Arch:
name: ResNet50_vd name: ResNet50_vd
pretrained: False pretrained: False
BackboneStopLayer: BackboneStopLayer:
name: flatten_0 name: "flatten"
Neck: Neck:
name: FC name: FC
embedding_size: 2048 embedding_size: 2048
......
...@@ -23,7 +23,7 @@ Arch: ...@@ -23,7 +23,7 @@ Arch:
name: "ResNet50_last_stage_stride1" name: "ResNet50_last_stage_stride1"
pretrained: True pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: "adaptive_avg_pool2d_0" name: "avg_pool"
Neck: Neck:
name: "VehicleNeck" name: "VehicleNeck"
in_channels: 2048 in_channels: 2048
......
...@@ -24,7 +24,7 @@ Arch: ...@@ -24,7 +24,7 @@ Arch:
name: "ResNet50_last_stage_stride1" name: "ResNet50_last_stage_stride1"
pretrained: True pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: "adaptive_avg_pool2d_0" name: "avg_pool"
Neck: Neck:
name: "VehicleNeck" name: "VehicleNeck"
in_channels: 2048 in_channels: 2048
......
...@@ -28,7 +28,7 @@ Arch: ...@@ -28,7 +28,7 @@ Arch:
name: "ResNet50_last_stage_stride1" name: "ResNet50_last_stage_stride1"
pretrained: True pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: "adaptive_avg_pool2d_0" name: "avg_pool"
Neck: Neck:
name: "VehicleNeck" name: "VehicleNeck"
in_channels: 2048 in_channels: 2048
......
...@@ -27,7 +27,7 @@ Arch: ...@@ -27,7 +27,7 @@ Arch:
name: "ResNet50_last_stage_stride1" name: "ResNet50_last_stage_stride1"
pretrained: True pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: "adaptive_avg_pool2d_0" name: "avg_pool"
Neck: Neck:
name: "VehicleNeck" name: "VehicleNeck"
in_channels: 2048 in_channels: 2048
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册