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

fix: replace reg exp with layer name

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