config.yml 1004 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. DATA:
  2. DIR: "data/training/"
  3. VAL_DIR: "data/training/"
  4. TEST_DIR: "data/NKL"
  5. LABEL_FILE: "data/training/nkl_train.txt"
  6. VAL_LABEL_FILE: "data/training/nkl_val.txt"
  7. TEST_LABEL_FILE: "data/training/nkl_test.txt"
  8. # DIR: "data/training/"
  9. # VAL_DIR: "data/training/"
  10. # TEST_DIR: "data/training"
  11. # LABEL_FILE: "data/training/train_1716_100_100.txt"
  12. # VAL_LABEL_FILE: "data/training/test_1716_100_100.txt"
  13. # TEST_LABEL_FILE: "data/training/test_1716_100_100.txt"
  14. BATCH_SIZE: 8
  15. WORKERS: 4
  16. OPTIMIZER:
  17. LR: 2e-4
  18. MOMENTUM: 0.9
  19. GAMMA: 0.1
  20. WEIGHT_DECAY: 0
  21. STEPS: []
  22. MODEL:
  23. NUMANGLE: 100
  24. NUMRHO: 100
  25. FIX: True
  26. THRESHOLD: 0.01
  27. EDGE_ALIGN: False
  28. BACKBONE: resnet50
  29. # only support : resnet18, resnet50, resnet101, mobilenetv2, resnext50, res2net50, vgg16
  30. TRAIN:
  31. EPOCHS: 30
  32. PRINT_FREQ: 100
  33. TEST: False
  34. SEED: 1997
  35. GPU_ID: 0
  36. DATA_PARALLEL: False
  37. RESUME:
  38. MISC:
  39. TMP: "./result/reproduce"