Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
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.
enlin
adeddaab09
|
2 years ago | |
|---|---|---|
| .. | ||
| README.md | 2 years ago | |
| __init__.py | 2 years ago | |
| datasets.py | 2 years ago | |
| macro.py | 2 years ago | |
| micro.py | 2 years ago | |
| mutator.py | 2 years ago | |
| ops.py | 2 years ago | |
| retrain.py | 2 years ago | |
| retrainer.py | 2 years ago | |
| search.py | 2 years ago | |
| selector.py | 2 years ago | |
| trainer.py | 2 years ago | |
| utils.py | 2 years ago | |
README.md
Efficient Neural Architecture Search (ENAS)
1. Requirements
torch
torchvision
collections
argparser
pickle
pytest-shutil
2.Train
Stage1: search an architecture
- macro search
python trainer.py --trial_id=0 --search_for macro --best_selected_space_path='./macro_selected_space.json' --result_path='./macro_result.json'
- micro search
python trainer.py --trial_id=0 --search_for micro --best_selected_space_path='./micro_selected_space.json' --result_path='./micro_result.json'
Stage2: select (deprecated)
python selector.py
stage3: retrain
- macro search
python retrainer.py --search_for macro --best_checkpoint_dir='./macro_checkpoint.pth' --best_selected_space_path=
'./macro_selected_space.json' --result_path='./macro_result.json'
- micro search
python retrainer.py --search_for micro --best_checkpoint_dir='./micro_checkpoint.pth' --best_selected_space_path=
'./micro_selected_space.json' --result_path='./micro_result.json'
一站式算法开发平台、高性能分布式深度学习框架、先进算法模型库、视觉模型炼知平台、数据可视化分析平台等一系列平台及工具，在模型高效分布式训练、数据处理和可视分析、模型炼知和轻量化等技术上形成独特优势，目前已在产学研等各领域近千家单位及个人提供AI应用赋能
Java Vue Python Text JavaScript other