Skip to content

Instantly share code, notes, and snippets.

@Taehun
Created July 27, 2021 05:45
Show Gist options
  • Save Taehun/07bfcf9ec34184db594cfdfb30d71195 to your computer and use it in GitHub Desktop.
Save Taehun/07bfcf9ec34184db594cfdfb30d71195 to your computer and use it in GitHub Desktop.
DDRNet23_slim default configurations
CUDNN:
BENCHMARK: true
DETERMINISTIC: false
ENABLED: true
GPUS: (0,1)
OUTPUT_DIR: 'output'
LOG_DIR: 'log'
WORKERS: 4
PRINT_FREQ: 10
DATASET:
DATASET: cityscapes
ROOT: data/
TEST_SET: 'list/cityscapes/val.lst'
TRAIN_SET: 'list/cityscapes/train.lst'
NUM_CLASSES: 19
MODEL:
NAME: ddrnet_23_slim
NUM_OUTPUTS: 2
PRETRAINED: "pretrained_models/DDRNet23s_imagenet.pth"
ALIGN_CORNERS: false
LOSS:
USE_OHEM: true
OHEMTHRES: 0.9
OHEMKEEP: 131072
BALANCE_WEIGHTS: [1, 0.4]
TRAIN:
IMAGE_SIZE:
- 1024
- 1024
BASE_SIZE: 2048
BATCH_SIZE_PER_GPU: 8
SHUFFLE: true
BEGIN_EPOCH: 0
END_EPOCH: 484
RESUME: false
OPTIMIZER: sgd
LR: 0.01
WD: 0.0005
MOMENTUM: 0.9
NESTEROV: false
FLIP: true
MULTI_SCALE: true
DOWNSAMPLERATE: 1
IGNORE_LABEL: 255
SCALE_FACTOR: 16
TEST:
IMAGE_SIZE:
- 2048
- 1024
BASE_SIZE: 2048
BATCH_SIZE_PER_GPU: 4
FLIP_TEST: false
MULTI_SCALE: false
SCALE_LIST: [1]
#0.5,0.75,1.0,1.25,1.5,1.75
# MODEL_FILE: "pretrained_models/best_val_smaller.pth"
OUTPUT_INDEX: 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment