mirror of https://github.com/JDAI-CV/fast-reid.git
Implement online pcb config
parent
44a90ae025
commit
d03eae9daa
|
@ -1,3 +1,5 @@
|
||||||
|
_BASE_: base.yaml
|
||||||
|
|
||||||
MODEL:
|
MODEL:
|
||||||
META_ARCHITECTURE: PCB
|
META_ARCHITECTURE: PCB
|
||||||
|
|
||||||
|
@ -37,61 +39,18 @@ MODEL:
|
||||||
SCALE: 1.
|
SCALE: 1.
|
||||||
|
|
||||||
INPUT:
|
INPUT:
|
||||||
SIZE_TRAIN: [480, 320]
|
SIZE_TRAIN: [310, 300]
|
||||||
SIZE_TEST: [480, 320]
|
SIZE_TEST: [310, 300]
|
||||||
|
|
||||||
FLIP:
|
CROP:
|
||||||
ENABLED: True
|
ENABLED: True
|
||||||
|
SIZE: [270, 260]
|
||||||
CJ:
|
SCALE: [0.8, 1.2]
|
||||||
ENABLED: True
|
RATIO: [3./4, 4./3]
|
||||||
|
|
||||||
AUTOAUG:
|
|
||||||
ENABLED: True
|
|
||||||
PROB: 0.5
|
|
||||||
|
|
||||||
AUGMIX:
|
|
||||||
ENABLED: True
|
|
||||||
PROB: 0.5
|
|
||||||
|
|
||||||
REA:
|
|
||||||
ENABLED: True
|
|
||||||
|
|
||||||
DATALOADER:
|
|
||||||
SAMPLER_TRAIN: TrainingSampler
|
|
||||||
NUM_WORKERS: 20
|
|
||||||
|
|
||||||
SOLVER:
|
|
||||||
MAX_EPOCH: 300
|
|
||||||
AMP:
|
|
||||||
ENABLED: True
|
|
||||||
|
|
||||||
OPT: SGD
|
|
||||||
SCHED: CosineAnnealingLR
|
|
||||||
|
|
||||||
BASE_LR: 0.01
|
|
||||||
MOMENTUM: 0.95
|
|
||||||
NESTEROV: True
|
|
||||||
|
|
||||||
BIAS_LR_FACTOR: 1.
|
|
||||||
WEIGHT_DECAY: 0.0005
|
|
||||||
WEIGHT_DECAY_BIAS: 0.
|
|
||||||
IMS_PER_BATCH: 96
|
|
||||||
|
|
||||||
ETA_MIN_LR: 0.00003
|
|
||||||
|
|
||||||
WARMUP_FACTOR: 0.1
|
|
||||||
WARMUP_ITERS: 100
|
|
||||||
|
|
||||||
CHECKPOINT_PERIOD: 1
|
|
||||||
|
|
||||||
TEST:
|
|
||||||
EVAL_PERIOD: 1
|
|
||||||
IMS_PER_BATCH: 128
|
|
||||||
|
|
||||||
DATASETS:
|
DATASETS:
|
||||||
NAMES: ("ShoeDataset",)
|
NAMES: ("ShoeDataset",)
|
||||||
TESTS: ("ShoeDataset", "OnlineDataset")
|
TESTS: ("ShoeDataset", "OnlineDataset")
|
||||||
|
|
||||||
OUTPUT_DIR: projects/FastShoe/logs/base-pair
|
OUTPUT_DIR: projects/FastShoe/logs/online-pcb
|
||||||
|
|
Loading…
Reference in New Issue