forked from Lausannen/NAS-FCOS
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathR_50_NAS_densebox.yaml
43 lines (43 loc) · 982 Bytes
/
R_50_NAS_densebox.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
MODEL:
META_ARCHITECTURE: "SingleStageDetector"
WEIGHT: "models/R-50-NAS.pth"
RPN_ONLY: True
DENSEBOX_ON: True
BACKBONE:
CONV_BODY: "R-50-NAS"
OUT_CHANNELS: 256
ENCODER_OUT_CHANNELS: (256, 512, 1024, 2048)
RETINANET:
SCALES_PER_OCTAVE: 3
STRADDLE_THRESH: -1
NUM_CLASSES: 81
NMS_TH: 0.6
USE_C5: False
DATASETS:
TRAIN: ("coco_2017_train",)
TEST: ("coco_2017_val",)
INPUT:
MIN_SIZE_TRAIN: (800,)
MAX_SIZE_TRAIN: 1333
MIN_SIZE_TEST: 800
MAX_SIZE_TEST: 1333
DATALOADER:
SIZE_DIVISIBILITY: 32
NUM_WORKERS: 2
SEARCH:
NAS_DECODER_ON: True
DECODER:
VERSION: 2
CONFIG: [[[3, 3, 4, 4, 1]], [[3, 3, 4, 4, 1]], [[4, 3, 4, 3, 1]], [[1, 3, 4, 3, 1]], [[5, 3, 4, 4, 1]], [[4, 7, 4, 4, 1]], [[7, 2, 4, 4, 1]]]
AGG_SIZE: 256
NUM_CELLS: 7
SOLVER:
BASE_LR: 0.01
WEIGHT_DECAY: 0.0001
STEPS: (60000,80000,)
MAX_ITER: 90000
IMS_PER_BATCH: 16
NUM_GPUS: 4
TEST:
IMS_PER_BATCH: 8
OUTPUT_DIR: 'output/coco/r50-NAS'