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.

readme 2.0 kB

2 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. stage1:python trainer.py --trial_id=1 --model_selected_space_path='./exp/train/2/model_selected_space.json' --search_space_path='./search_space.json' --result_path='./exp/train/2/result.json'
  2. stage2:python selector.py --experiment_dir='./exp' --best_selected_space_path='./best_selected_space.json'
  3. stage3:python retrainer.py --best_checkpoint_dir='experiment_id/' --best_selected_space_path='./best_selected_space.json' --result_path='result.json'
  4. search_space.json:
  5. {
  6. "first_conv": {
  7. "_type": "layer_choice",
  8. "_value": [
  9. "conv5x5",
  10. "conv3x3"
  11. ]
  12. },
  13. "mid_conv": {
  14. "_type": "layer_choice",
  15. "_value": [
  16. "0",
  17. "1"
  18. ]
  19. },
  20. "skip": {
  21. "_type": "input_choice",
  22. "_value": {
  23. "candidates": [
  24. "",
  25. ""
  26. ],
  27. "n_chosen": 1
  28. }
  29. }
  30. }
  31. selected_space.json:
  32. {
  33. "first_conv": {
  34. "_idx": 0,
  35. "_value": "conv5x5"
  36. },
  37. "mid_conv": {
  38. "_idx": 0,
  39. "_value": "0"
  40. },
  41. "skip": {
  42. "_idx": [
  43. 0
  44. ],
  45. "_value": [
  46. ""
  47. ]
  48. }
  49. }
  50. result.json:
  51. {'type': 'accuracy', 'result': {'sequence': 1, 'category': 'epoch', 'value': 96.73815907059875}}
  52. {'type': 'accuracy', 'result': {'sequence': 2, 'category': 'epoch', 'value': 97.6988382484361}}
  53. {'type': 'accuracy', 'result': {'sequence': 3, 'category': 'epoch', 'value': 98.63717605004469}}
  54. {'type': 'accuracy', 'result': {'sequence': 4, 'category': 'epoch', 'value': 98.72654155495978}}
  55. {'type': 'accuracy', 'result': {'sequence': 5, 'category': 'epoch', 'value': 99.27390527256479}}
  56. {'type': 'accuracy', 'result': {'sequence': 6, 'category': 'epoch', 'value': 99.13985701519213}}
  57. {'type': 'accuracy', 'result': {'sequence': 7, 'category': 'epoch', 'value': 99.3632707774799}}
  58. {'type': 'accuracy', 'result': {'sequence': 8, 'category': 'epoch', 'value': 99.4414655942806}}
  59. {'type': 'accuracy', 'result': {'sequence': 9, 'category': 'epoch', 'value': 99.67605004468275}}
  60. {'type': 'accuracy', 'result': {'sequence': 10, 'category': 'epoch', 'value': 99.74307417336908}}

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