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

Simplify configurations a bit (#2510)

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