mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

auto port in dist_url

pull/608/head
zuchen.wang 2021-11-12 20:21:40 +08:00
parent bc63002d05
commit 339b5f2188
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
projects/FastShoe/configs/online-pcb.yaml
View File

@ -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",)

2
projects/FastShoe/train_net.py
View File

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