From d4a3e11ab2be3cb56f670361e31e66a086128f67 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 17 Oct 2020 08:28:37 -0600 Subject: [PATCH] Don't use several stages of spinenet_arch These are used for lower outputs which I am not using --- codes/models/archs/spinenet_arch.py | 4 ---- codes/train2.py | 2 +- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/codes/models/archs/spinenet_arch.py b/codes/models/archs/spinenet_arch.py index 99f95fa7..86bac6f4 100644 --- a/codes/models/archs/spinenet_arch.py +++ b/codes/models/archs/spinenet_arch.py @@ -93,12 +93,8 @@ SPINENET_BLOCK_SPECS = [ (5, BasicBlock, (6, 7), False), (7, BasicBlock, (6, 8), False), (5, Bottleneck, (8, 9), False), - (5, Bottleneck, (8, 10), False), (4, Bottleneck, (5, 10), True), (3, Bottleneck, (4, 10), True), - (5, Bottleneck, (7, 12), True), - (7, Bottleneck, (5, 14), True), - (6, Bottleneck, (12, 14), True), ] SCALING_MAP = { diff --git a/codes/train2.py b/codes/train2.py index 396ab01d..f38eee3e 100644 --- a/codes/train2.py +++ b/codes/train2.py @@ -30,7 +30,7 @@ def init_dist(backend='nccl', **kwargs): def main(): #### options parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_chained_structured.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_chained_constrained.yml') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--local_rank', type=int, default=0) args = parser.parse_args()