| GPUS: (0,1,2,3) | |
| LOG_DIR: 'log/' | |
| DATA_DIR: '' | |
| OUTPUT_DIR: 'output/' | |
| WORKERS: 4 | |
| PRINT_FREQ: 1000 | |
| MODEL: | |
| NAME: cls_hrnet | |
| IMAGE_SIZE: | |
| - 224 | |
| - 224 | |
| EXTRA: | |
| STAGE1: | |
| NUM_MODULES: 1 | |
| NUM_RANCHES: 1 | |
| BLOCK: BOTTLENECK | |
| NUM_BLOCKS: | |
| - 4 | |
| NUM_CHANNELS: | |
| - 64 | |
| FUSE_METHOD: SUM | |
| STAGE2: | |
| NUM_MODULES: 1 | |
| NUM_BRANCHES: 2 | |
| BLOCK: BASIC | |
| NUM_BLOCKS: | |
| - 4 | |
| - 4 | |
| NUM_CHANNELS: | |
| - 64 | |
| - 128 | |
| FUSE_METHOD: SUM | |
| STAGE3: | |
| NUM_MODULES: 4 | |
| NUM_BRANCHES: 3 | |
| BLOCK: BASIC | |
| NUM_BLOCKS: | |
| - 4 | |
| - 4 | |
| - 4 | |
| NUM_CHANNELS: | |
| - 64 | |
| - 128 | |
| - 256 | |
| FUSE_METHOD: SUM | |
| STAGE4: | |
| NUM_MODULES: 3 | |
| NUM_BRANCHES: 4 | |
| BLOCK: BASIC | |
| NUM_BLOCKS: | |
| - 4 | |
| - 4 | |
| - 4 | |
| - 4 | |
| NUM_CHANNELS: | |
| - 64 | |
| - 128 | |
| - 256 | |
| - 512 | |
| FUSE_METHOD: SUM | |
| CUDNN: | |
| BENCHMARK: true | |
| DETERMINISTIC: false | |
| ENABLED: true | |
| DATASET: | |
| DATASET: 'imagenet' | |
| DATA_FORMAT: 'jpg' | |
| ROOT: 'data/imagenet/' | |
| TEST_SET: 'val' | |
| TRAIN_SET: 'train' | |
| TEST: | |
| BATCH_SIZE_PER_GPU: 32 | |
| MODEL_FILE: '' | |
| TRAIN: | |
| BATCH_SIZE_PER_GPU: 32 | |
| BEGIN_EPOCH: 0 | |
| END_EPOCH: 100 | |
| RESUME: true | |
| LR_FACTOR: 0.1 | |
| LR_STEP: | |
| - 30 | |
| - 60 | |
| - 90 | |
| OPTIMIZER: sgd | |
| LR: 0.05 | |
| WD: 0.0001 | |
| MOMENTUM: 0.9 | |
| NESTEROV: true | |
| SHUFFLE: true | |
| DEBUG: | |
| DEBUG: false | |