diff --git a/projects/FastShoe/configs/online-pcb.yaml b/projects/FastShoe/configs/online-pcb.yaml index 0afd488..420e4ae 100644 --- a/projects/FastShoe/configs/online-pcb.yaml +++ b/projects/FastShoe/configs/online-pcb.yaml @@ -49,7 +49,7 @@ INPUT: RATIO: [0.75, 1.33333333] DATALOADER: - NUM_WORKERS: 8 + NUM_WORKERS: 16 SOLVER: AMP: @@ -58,23 +58,23 @@ SOLVER: OPT: SGD SCHED: MultiStepLR - BASE_LR: 0.001 - MOMENTUM: 0.9 + BASE_LR: 0.0005 + MOMENTUM: 0.99 NESTEROV: True BIAS_LR_FACTOR: 1. - WEIGHT_DECAY: 0.0005 + WEIGHT_DECAY: 0.00001 WEIGHT_DECAY_BIAS: 0. - ETA_MIN_LR: 0.00003 + ETA_MIN_LR: 0.000003 WARMUP_FACTOR: 0.1 WARMUP_ITERS: 0 - MAX_EPOCH: 100 - IMS_PER_BATCH: 80 + MAX_EPOCH: 300 + IMS_PER_BATCH: 168 TEST: - IMS_PER_BATCH: 480 + IMS_PER_BATCH: 336 DATASETS: NAMES: ("PairDataset",) diff --git a/projects/FastShoe/train_net.py b/projects/FastShoe/train_net.py index cb21ee8..de2599c 100644 --- a/projects/FastShoe/train_net.py +++ b/projects/FastShoe/train_net.py @@ -55,6 +55,6 @@ if __name__ == "__main__": args.num_gpus, num_machines=args.num_machines, machine_rank=args.machine_rank, - dist_url=args.dist_url, + dist_url='auto', args=(args,), )