MobileNetV3_large_x1_0_finetune.yaml 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. mode: 'train'
  2. ARCHITECTURE:
  3. name: 'MobileNetV3_large_x1_0'
  4. pretrained_model: "/home/xiongweixp/Documents/vscode_project/paddle-handwrite/output_2/MobileNetV3_large_x1_0_pretrained"
  5. model_save_dir: "/home/xiongweixp/Documents/vscode_project/paddle-handwrite/output_2"
  6. use_gpu: True
  7. classes_num: 3755
  8. total_images: 631381
  9. save_interval: 1
  10. validate: True
  11. valid_interval: 1
  12. epochs: 20
  13. topk: 5
  14. image_shape: [3, 224, 224]
  15. LEARNING_RATE:
  16. function: 'Cosine'
  17. params:
  18. lr: 0.00375
  19. OPTIMIZER:
  20. function: 'Momentum'
  21. params:
  22. momentum: 0.9
  23. regularizer:
  24. function: 'L2'
  25. factor: 0.000001
  26. TRAIN:
  27. batch_size: 32
  28. num_workers: 0
  29. file_list: "/home/xiongweixp/Documents/vscode_project/paddle-handwrite/data/trade/train_list.txt"
  30. data_dir: "/home/xiongweixp/Documents/vscode_project/paddle-handwrite/data/trade/"
  31. shuffle_seed: 0
  32. transforms:
  33. - DecodeImage:
  34. to_rgb: True
  35. to_np: False
  36. channel_first: False
  37. - RandCropImage:
  38. size: 224
  39. - RandFlipImage:
  40. flip_code: 1
  41. - NormalizeImage:
  42. scale: 1./255.
  43. mean: [0.485, 0.456, 0.406]
  44. std: [0.229, 0.224, 0.225]
  45. order: ''
  46. - ToCHWImage:
  47. VALID:
  48. batch_size: 20
  49. num_workers: 0
  50. file_list: "/home/xiongweixp/Documents/vscode_project/paddle-handwrite/data/trade/val_list.txt"
  51. data_dir: "/home/xiongweixp/Documents/vscode_project/paddle-handwrite/data/trade/"
  52. shuffle_seed: 0
  53. transforms:
  54. - DecodeImage:
  55. to_rgb: True
  56. to_np: False
  57. channel_first: False
  58. - ResizeImage:
  59. resize_short: 256
  60. - CropImage:
  61. size: 224
  62. - NormalizeImage:
  63. scale: 1.0/255.0
  64. mean: [0.485, 0.456, 0.406]
  65. std: [0.229, 0.224, 0.225]
  66. order: ''
  67. - ToCHWImage: