OWN_config.yaml 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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: OWN
  13. ROOT: "H:/DL-DATASET/360M/images"
  14. JSON_FILE: {'train': 'lib/dataset/txt/train_own.txt', 'val': 'lib/dataset/txt/test_own.txt'}
  15. SCALE_FACTOR: 0.25
  16. ROT_FACTOR: 30
  17. STD: 0.193
  18. MEAN: 0.588
  19. ALPHABETS: ''
  20. TRAIN:
  21. BATCH_SIZE_PER_GPU: 32
  22. SHUFFLE: True
  23. BEGIN_EPOCH: 0
  24. END_EPOCH: 100
  25. RESUME:
  26. IS_RESUME: False
  27. FILE: ''
  28. OPTIMIZER: 'adam'
  29. LR: 0.0001
  30. WD: 0.0
  31. LR_STEP: [60, 80]
  32. LR_FACTOR: 0.1
  33. MOMENTUM: 0.0
  34. NESTEROV: False
  35. RMSPROP_ALPHA:
  36. RMSPROP_CENTERED:
  37. FINETUNE:
  38. IS_FINETUNE: true
  39. FINETUNE_CHECKPOINIT: 'output/checkpoints/mixed_second_finetune_acc_97P7.pth'
  40. FREEZE: true
  41. TEST:
  42. BATCH_SIZE_PER_GPU: 16
  43. SHUFFLE: True # for random test rather than test on the whole validation set
  44. NUM_TEST_BATCH: 1000
  45. NUM_TEST_DISP: 10
  46. MODEL:
  47. NAME: 'crnn'
  48. IMAGE_SIZE:
  49. OW: 280. # origial width: 280
  50. H: 32
  51. W: 160 # resized width: 160
  52. NUM_CLASSES: 0
  53. NUM_HIDDEN: 256