提交 62d2f3d2 编写于 作者: Y Yang Zhang 提交者: qingqing01

Simplify configurations a bit (#2510)

上级 7a13a4b8
...@@ -26,7 +26,6 @@ ResNet: ...@@ -26,7 +26,6 @@ ResNet:
freeze_at: 2 freeze_at: 2
variant: b variant: b
FPN: FPN:
min_level: 2 min_level: 2
max_level: 6 max_level: 6
...@@ -87,6 +86,7 @@ CascadeBBoxHead: ...@@ -87,6 +86,7 @@ CascadeBBoxHead:
FC6FC7Head: FC6FC7Head:
num_chan: 1024 num_chan: 1024
LearningRate: LearningRate:
base_lr: 0.02 base_lr: 0.02
schedulers: schedulers:
......
...@@ -21,7 +21,7 @@ FasterRCNN: ...@@ -21,7 +21,7 @@ FasterRCNN:
ResNet: ResNet:
norm_type: affine_channel norm_type: affine_channel
depth: 101 depth: 101
feature_maps: [2,3,4] feature_maps: 4
freeze_at: 2 freeze_at: 2
ResNetC5: ResNetC5:
...@@ -117,4 +117,3 @@ FasterRCNNTestFeed: ...@@ -117,4 +117,3 @@ FasterRCNNTestFeed:
image_dir: val2017 image_dir: val2017
test_file: val2017.txt test_file: val2017.txt
samples: 5 samples: 5
...@@ -21,7 +21,7 @@ FasterRCNN: ...@@ -21,7 +21,7 @@ FasterRCNN:
ResNet: ResNet:
norm_type: affine_channel norm_type: affine_channel
depth: 50 depth: 50
feature_maps: [2,3,4] feature_maps: 4
freeze_at: 2 freeze_at: 2
ResNetC5: ResNetC5:
......
...@@ -21,7 +21,7 @@ FasterRCNN: ...@@ -21,7 +21,7 @@ FasterRCNN:
ResNet: ResNet:
norm_type: affine_channel norm_type: affine_channel
depth: 50 depth: 50
feature_maps: [2,3,4] feature_maps: 4
freeze_at: 2 freeze_at: 2
ResNetC5: ResNetC5:
......
...@@ -21,7 +21,7 @@ FasterRCNN: ...@@ -21,7 +21,7 @@ FasterRCNN:
ResNet: ResNet:
norm_type: affine_channel norm_type: affine_channel
depth: 50 depth: 50
feature_maps: [2,3,4] feature_maps: 4
freeze_at: 2 freeze_at: 2
variant: d variant: d
......
...@@ -21,7 +21,7 @@ FasterRCNN: ...@@ -21,7 +21,7 @@ FasterRCNN:
SENet: SENet:
depth: 152 depth: 152
feature_maps: [2, 3, 4] feature_maps: 4
freeze_at: 2 freeze_at: 2
group_width: 4 group_width: 4
groups: 64 groups: 64
...@@ -30,7 +30,6 @@ SENet: ...@@ -30,7 +30,6 @@ SENet:
SENetC5: SENetC5:
depth: 152 depth: 152
feature_maps: 5
freeze_at: 2 freeze_at: 2
group_width: 4 group_width: 4
groups: 64 groups: 64
......
...@@ -25,7 +25,7 @@ ResNet: ...@@ -25,7 +25,7 @@ ResNet:
norm_type: affine_channel norm_type: affine_channel
norm_decay: true norm_decay: true
depth: 50 depth: 50
feature_maps: [2,3, 4] feature_maps: 4
freeze_at: 2 freeze_at: 2
ResNetC5: ResNetC5:
......
...@@ -25,7 +25,7 @@ ResNet: ...@@ -25,7 +25,7 @@ ResNet:
norm_type: affine_channel norm_type: affine_channel
norm_decay: true norm_decay: true
depth: 50 depth: 50
feature_maps: [2,3, 4] feature_maps: 4
freeze_at: 2 freeze_at: 2
ResNetC5: ResNetC5:
......
...@@ -49,7 +49,7 @@ def main(): ...@@ -49,7 +49,7 @@ def main():
merge_config(args.cli_config) merge_config(args.cli_config)
if 'test_feed' not in cfg: if 'test_feed' not in cfg:
test_feed = create(type(main_arch).__name__ + 'TestFeed') test_feed = create(main_arch + 'TestFeed')
else: else:
test_feed = create(cfg['test_feed']) test_feed = create(cfg['test_feed'])
......
...@@ -56,13 +56,13 @@ def main(): ...@@ -56,13 +56,13 @@ def main():
devices_num = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count())) devices_num = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count()))
if 'train_feed' not in cfg: if 'train_feed' not in cfg:
train_feed = create(type(main_arch).__name__ + 'TrainFeed') train_feed = create(main_arch + 'TrainFeed')
else: else:
train_feed = create(cfg['train_feed']) train_feed = create(cfg['train_feed'])
if args.eval: if args.eval:
if 'eval_feed' not in cfg: if 'eval_feed' not in cfg:
eval_feed = create(type(main_arch).__name__ + 'EvalFeed') eval_feed = create(main_arch + 'EvalFeed')
else: else:
eval_feed = create(cfg['eval_feed']) eval_feed = create(cfg['eval_feed'])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册