You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

62 lines
1.3KB

  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: "../textGenerator/dataset/dataset9/images"
  14. CHAR_FILE: '../textGenerator/dataset/dataset9/chars.txt'
  15. JSON_FILE: {'train': '../textGenerator/dataset/dataset9/train.txt', 'val': '../textGenerator/dataset/dataset9/val.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: 'output/360CC/crnn/2023-04-27-13-01/checkpoints/checkpoint_99_acc_0.5030.pth'
  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: False
  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 # for random test rather than test on the whole validation set
  46. NUM_TEST_BATCH: 1000
  47. NUM_TEST_DISP: 10
  48. MODEL:
  49. NAME: 'crnn'
  50. IMAGE_SIZE:
  51. OW: 160 # origial width: 280
  52. H: 32
  53. W: 160 # resized width: 160
  54. NUM_CLASSES: 0
  55. NUM_HIDDEN: 256