360CC_config.yaml 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. GPUID: 0
  2. WORKERS: 1
  3. PRINT_FREQ: 10
  4. SAVE_FREQ: 10
  5. PIN_MEMORY: False
  6. OUTPUT_DIR: 'output'
  7. CUDNN:
  8. BENCHMARK: True
  9. DETERMINISTIC: False
  10. ENABLED: True
  11. DATASET:
  12. DATASET: 360CC
  13. ROOT: "H:/DL-DATASET/360M/images"
  14. CHAR_FILE: 'lib/dataset/txt/char_std_5990.txt'
  15. JSON_FILE: {'train': 'lib/dataset/txt/train.txt', 'val': 'lib/dataset/txt/test.txt'}
  16. # JSON_FILE: {'train': 'H:/DL-DATASET/360M/train.txt', 'val': 'H:/DL-DATASET/360M/test.txt'}
  17. SCALE_FACTOR: 0.25
  18. ROT_FACTOR: 30
  19. STD: 0.193
  20. MEAN: 0.588
  21. ALPHABETS: ''
  22. TRAIN:
  23. BATCH_SIZE_PER_GPU: 32
  24. SHUFFLE: True
  25. BEGIN_EPOCH: 0
  26. END_EPOCH: 100
  27. RESUME:
  28. IS_RESUME: False
  29. FILE: ''
  30. OPTIMIZER: 'adam'
  31. LR: 0.0001
  32. WD: 0.0
  33. LR_STEP: [60, 80]
  34. LR_FACTOR: 0.1
  35. MOMENTUM: 0.0
  36. NESTEROV: False
  37. RMSPROP_ALPHA:
  38. RMSPROP_CENTERED:
  39. FINETUNE:
  40. IS_FINETUNE: true
  41. FINETUNE_CHECKPOINIT: 'output/checkpoints/mixed_second_finetune_acc_97P7.pth'
  42. FREEZE: true
  43. TEST:
  44. BATCH_SIZE_PER_GPU: 16
  45. SHUFFLE: True
  46. NUM_TEST: 1000
  47. NUM_TEST_DISP: 10
  48. MODEL:
  49. NAME: 'crnn'
  50. IMAGE_SIZE:
  51. OW: 280. # origial width: 280
  52. H: 32
  53. W: 160 # resized width: 160
  54. NUM_CLASSES: 0
  55. NUM_HIDDEN: 256