12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- DATA:
- DIR: "data/training/"
- VAL_DIR: "data/training/"
- TEST_DIR: "data/NKL"
- LABEL_FILE: "data/training/nkl_train.txt"
- VAL_LABEL_FILE: "data/training/nkl_val.txt"
- TEST_LABEL_FILE: "data/training/nkl_test.txt"
- # DIR: "data/training/"
- # VAL_DIR: "data/training/"
- # TEST_DIR: "data/training"
- # LABEL_FILE: "data/training/train_1716_100_100.txt"
- # VAL_LABEL_FILE: "data/training/test_1716_100_100.txt"
- # TEST_LABEL_FILE: "data/training/test_1716_100_100.txt"
- BATCH_SIZE: 8
- WORKERS: 4
- OPTIMIZER:
- LR: 2e-4
- MOMENTUM: 0.9
- GAMMA: 0.1
- WEIGHT_DECAY: 0
- STEPS: []
- MODEL:
- NUMANGLE: 100
- NUMRHO: 100
- FIX: True
- THRESHOLD: 0.01
- EDGE_ALIGN: False
- BACKBONE: resnet50
- # only support : resnet18, resnet50, resnet101, mobilenetv2, resnext50, res2net50, vgg16
- TRAIN:
- EPOCHS: 30
- PRINT_FREQ: 100
- TEST: False
- SEED: 1997
- GPU_ID: 0
- DATA_PARALLEL: False
- RESUME:
- MISC:
- TMP: "./result/reproduce"
|