eschmidbauer/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't use several stages of spinenet_arch

Browse Source 
These are used for lower outputs which I am not using
This commit is contained in:
James Betker 2020-10-17 08:28:37 -06:00
parent d1c63ae339
commit d4a3e11ab2
2 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/archs/spinenet_arch.py
View File

@ -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 = {

2
codes/train2.py
View File

@ -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()