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

reademe.md 2.4 kB

2 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. # train stage
  2. `python pcdarts_train.py --data_dir '../data/' --result_path 'trial_id/result.json' --log_path 'trial_id/log' --search_space_path 'experiment_id/search_space.json' --best_selected_space_path 'experiment_id/best_selected_space.json' --trial_id 0 --layers 5 --model_lr 0.025 --arch_lr 3e-4 --epochs 2 --pre_epochs 1 --batch_size 64 --channels 16`
  3. # select stage
  4. `python pcdarts_select.py --best_selected_space_path 'experiment_id/best_selected_space.json' `
  5. # retrain stage
  6. `python pcdarts_retrain.py --data_dir '../data/' --result_path 'trial_id/result.json' --log_path 'trial_id/log' --best_selected_space_path 'experiment_id/best_selected_space.json' --best_checkpoint_dir 'experiment_id/' --trial_id 0 --batch_size 96 --epochs 2 --lr 0.01 --layers 20 --channels 36`
  7. # output file
  8. `result.json`
  9. ```
  10. {'type': 'Accuracy', 'result': {'sequence': 0, 'category': 'epoch', 'value': 0.1}}
  11. {'type': 'Accuracy', 'result': {'sequence': 1, 'category': 'epoch', 'value': 0.0}}
  12. {'type': 'Accuracy', 'result': {'sequence': 2, 'category': 'epoch', 'value': 0.0}}
  13. {'type': 'Accuracy', 'result': {'sequence': 3, 'category': 'epoch', 'value': 0.0}}
  14. {'type': 'Accuracy', 'result': {'sequence': 4, 'category': 'epoch', 'value': 0.0}}
  15. {'type': 'Cost_time', 'result': {'value': '41.614346981048584 s'}}
  16. ```
  17. `search_space.json`
  18. ```
  19. {
  20. "op_list": {
  21. "_type": "layer_choice",
  22. "_value": [
  23. "maxpool",
  24. "avgpool",
  25. "skipconnect",
  26. "sepconv3x3",
  27. "sepconv5x5",
  28. "dilconv3x3",
  29. "dilconv5x5",
  30. "none"
  31. ]
  32. },
  33. "search_space": {
  34. "normal_n2_p0": "op_list",
  35. "normal_n2_p1": "op_list",
  36. "normal_n2_switch": {
  37. "_type": "input_choice",
  38. "_value": {
  39. "candidates": [
  40. "normal_n2_p0",
  41. "normal_n2_p1"
  42. ],
  43. "n_chosen": 2
  44. }
  45. },
  46. ...
  47. }
  48. ```
  49. `best_selected_space.json`
  50. ```
  51. {
  52. "normal_n2_p0": "dilconv5x5",
  53. "normal_n2_p1": "dilconv5x5",
  54. "normal_n2_switch": [
  55. "normal_n2_p0",
  56. "normal_n2_p1"
  57. ],
  58. "normal_n3_p0": "sepconv3x3",
  59. "normal_n3_p1": "dilconv5x5",
  60. "normal_n3_p2": [],
  61. "normal_n3_switch": [
  62. "normal_n3_p0",
  63. "normal_n3_p1"
  64. ],
  65. "normal_n4_p0": [],
  66. "normal_n4_p1": "dilconv5x5",
  67. "normal_n4_p2": "sepconv5x5",
  68. "normal_n4_p3": [],
  69. "normal_n4_switch": [
  70. "normal_n4_p1",
  71. "normal_n4_p2"
  72. ],
  73. ...
  74. }
  75. ```

一站式算法开发平台、高性能分布式深度学习框架、先进算法模型库、视觉模型炼知平台、数据可视化分析平台等一系列平台及工具,在模型高效分布式训练、数据处理和可视分析、模型炼知和轻量化等技术上形成独特优势,目前已在产学研等各领域近千家单位及个人提供AI应用赋能