2021-01-18 11:36:38 +08:00
|
|
|
MODEL:
|
|
|
|
META_ARCHITECTURE: Baseline
|
|
|
|
|
|
|
|
BACKBONE:
|
|
|
|
NAME: build_resnet_backbone
|
2021-04-02 21:33:13 +08:00
|
|
|
DEPTH: 18x
|
2021-01-18 11:36:38 +08:00
|
|
|
NORM: BN
|
|
|
|
LAST_STRIDE: 2
|
2021-04-02 21:33:13 +08:00
|
|
|
FEAT_DIM: 512
|
2021-01-18 11:36:38 +08:00
|
|
|
PRETRAIN: True
|
|
|
|
|
|
|
|
HEADS:
|
2021-03-26 20:17:39 +08:00
|
|
|
NAME: ClasHead
|
2021-01-18 11:36:38 +08:00
|
|
|
WITH_BNNECK: False
|
|
|
|
EMBEDDING_DIM: 0
|
2021-03-26 20:17:39 +08:00
|
|
|
POOL_LAYER: FastGlobalAvgPool
|
|
|
|
CLS_LAYER: Linear
|
|
|
|
NUM_CLASSES: 2
|
2021-01-18 11:36:38 +08:00
|
|
|
|
|
|
|
LOSSES:
|
|
|
|
NAME: ("CrossEntropyLoss",)
|
|
|
|
|
|
|
|
CE:
|
|
|
|
EPSILON: 0.1
|
|
|
|
SCALE: 1.
|
|
|
|
|
|
|
|
INPUT:
|
2021-04-02 21:33:13 +08:00
|
|
|
SIZE_TRAIN: [0,] # no need for resize when training
|
|
|
|
SIZE_TEST: [256,]
|
2021-03-31 17:07:19 +08:00
|
|
|
|
2021-03-26 20:17:39 +08:00
|
|
|
CROP:
|
|
|
|
ENABLED: True
|
2021-04-02 21:33:13 +08:00
|
|
|
SIZE: [224,]
|
|
|
|
SCALE: [0.08, 1]
|
|
|
|
RATIO: [0.75, 1.333333333]
|
2021-03-31 17:07:19 +08:00
|
|
|
|
2021-03-26 20:17:39 +08:00
|
|
|
FLIP:
|
|
|
|
ENABLED: True
|
|
|
|
|
2021-01-18 11:36:38 +08:00
|
|
|
DATALOADER:
|
2021-03-26 20:17:39 +08:00
|
|
|
SAMPLER_TRAIN: TrainingSampler
|
2021-01-18 11:36:38 +08:00
|
|
|
NUM_WORKERS: 8
|
|
|
|
|
|
|
|
SOLVER:
|
|
|
|
MAX_EPOCH: 100
|
2021-03-26 20:17:39 +08:00
|
|
|
AMP:
|
|
|
|
ENABLED: True
|
2021-01-18 11:36:38 +08:00
|
|
|
|
|
|
|
OPT: SGD
|
|
|
|
SCHED: CosineAnnealingLR
|
|
|
|
|
2021-04-02 21:33:13 +08:00
|
|
|
BASE_LR: 0.001
|
|
|
|
MOMENTUM: 0.9
|
|
|
|
NESTEROV: False
|
2021-01-18 11:36:38 +08:00
|
|
|
|
|
|
|
BIAS_LR_FACTOR: 1.
|
|
|
|
WEIGHT_DECAY: 0.0005
|
|
|
|
WEIGHT_DECAY_BIAS: 0.
|
2021-07-05 11:10:37 +08:00
|
|
|
IMS_PER_BATCH: 16
|
2021-01-18 11:36:38 +08:00
|
|
|
|
|
|
|
ETA_MIN_LR: 0.00003
|
|
|
|
|
|
|
|
WARMUP_FACTOR: 0.1
|
2021-04-02 21:33:13 +08:00
|
|
|
WARMUP_ITERS: 100
|
2021-01-18 11:36:38 +08:00
|
|
|
|
|
|
|
CHECKPOINT_PERIOD: 10
|
|
|
|
|
|
|
|
TEST:
|
|
|
|
EVAL_PERIOD: 10
|
|
|
|
IMS_PER_BATCH: 256
|
|
|
|
|
|
|
|
DATASETS:
|
|
|
|
NAMES: ("Hymenoptera",)
|
|
|
|
TESTS: ("Hymenoptera",)
|
|
|
|
|
2021-04-02 21:33:13 +08:00
|
|
|
OUTPUT_DIR: projects/FastClas/logs/r18_demo
|