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

73 lines
1.2 KiB
YAML
Raw Normal View History

style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
MODEL:
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
META_ARCHITECTURE: "Baseline"
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
BACKBONE:
NAME: "build_resnet_backbone"
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
NORM: "BN"
onnx/trt support Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
DEPTH: "50x"
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
LAST_STRIDE: 1
change config file
2020-09-10 10:58:56 +08:00
FEAT_DIM: 2048
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
WITH_IBN: False
PRETRAIN: True
change config file
2020-09-10 10:58:56 +08:00
PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50-19c8e357.pth"
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
HEADS:
change config file
2020-09-10 10:58:56 +08:00
NAME: "EmbeddingHead"
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
NORM: "BN"
change config file
2020-09-10 10:58:56 +08:00
WITH_BNNECK: True
update all config files
2020-07-06 16:48:14 +08:00
POOL_LAYER: "avgpool"
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
NECK_FEAT: "before"
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
CLS_LAYER: "linear"
LOSSES:
update all config files
2020-07-06 16:48:14 +08:00
NAME: ("CrossEntropyLoss", "TripletLoss",)
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
CE:
EPSILON: 0.1
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
SCALE: 1.
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
TRI:
chore(configs): update Market1501 training config
2020-04-29 21:06:57 +08:00
MARGIN: 0.3
HARD_MINING: True
update all config files
2020-07-06 16:48:14 +08:00
NORM_FEAT: False
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
SCALE: 1.
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
INPUT:
chore(configs): update Market1501 training config
2020-04-29 21:06:57 +08:00
SIZE_TRAIN: [256, 128]
SIZE_TEST: [256, 128]
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
REA:
ENABLED: True
PROB: 0.5
DO_PAD: True
DATALOADER:
PK_SAMPLER: True
update all config files
2020-07-06 16:48:14 +08:00
NAIVE_WAY: True
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
NUM_INSTANCE: 4
update all config files
2020-07-06 16:48:14 +08:00
NUM_WORKERS: 8
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
SOLVER:
v0.3 update Summary: 1. change DPP training in apex way; 2. make warmup scheduler by iter and lr scheduler by epoch; 3. replace random erasing with torchvision implementation; 4. naming modification in config file
2020-12-07 14:19:20 +08:00
FP16_ENABLED: True
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
OPT: "Adam"
v0.3 update Summary: 1. change DPP training in apex way; 2. make warmup scheduler by iter and lr scheduler by epoch; 3. replace random erasing with torchvision implementation; 4. naming modification in config file
2020-12-07 14:19:20 +08:00
MAX_EPOCH: 120
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
BASE_LR: 0.00035
BIAS_LR_FACTOR: 2.
WEIGHT_DECAY: 0.0005
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
WEIGHT_DECAY_BIAS: 0.0005
chore(configs): update Market1501 training config
2020-04-29 21:06:57 +08:00
IMS_PER_BATCH: 64
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
v0.3 update Summary: 1. change DPP training in apex way; 2. make warmup scheduler by iter and lr scheduler by epoch; 3. replace random erasing with torchvision implementation; 4. naming modification in config file
2020-12-07 14:19:20 +08:00
SCHED: "MultiStepLR"
update all config files
2020-07-06 16:48:14 +08:00
STEPS: [40, 90]
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
GAMMA: 0.1
v0.3 update Summary: 1. change DPP training in apex way; 2. make warmup scheduler by iter and lr scheduler by epoch; 3. replace random erasing with torchvision implementation; 4. naming modification in config file
2020-12-07 14:19:20 +08:00
WARMUP_FACTOR: 0.1
WARMUP_ITERS: 2000
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
v0.3 update Summary: 1. change DPP training in apex way; 2. make warmup scheduler by iter and lr scheduler by epoch; 3. replace random erasing with torchvision implementation; 4. naming modification in config file
2020-12-07 14:19:20 +08:00
CHECKPOINT_PERIOD: 30
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
TEST:
update all config files
2020-07-06 16:48:14 +08:00
EVAL_PERIOD: 30
IMS_PER_BATCH: 128
style(configs): put all config files together put all config files into one place for easily control, and add tools for put train_net.py which almost the same in different projects
2020-04-29 16:18:54 +08:00
CUDNN_BENCHMARK: True