hummingbird
/
Dubhe
Not watched
2
0
Fork 0
Code
Releases 5 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
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.
129 Commits
3 Branches
639 MB
3 Download
 
 
 
 
 
 
Tree: 7d48472254
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7d48472254'
${ noResults }
Dubhe/dubhe-tadl/cream
enlin adeddaab09 add tadl算法 2 years ago
..
algorithms add tadl算法 2 years ago
lib add tadl算法 2 years ago
pretrained add tadl算法 2 years ago
README.md add tadl算法 2 years ago
retrainer.py add tadl算法 2 years ago
selector.py add tadl算法 2 years ago
test.py add tadl算法 2 years ago
trainer.py add tadl算法 2 years ago

README.md

Cream of the Crop: Distilling Prioritized Paths For One-Shot Neural Architecture Search

0x01 requirements

  • Install the following requirements:
future
thop
timm<0.4
yacs
ptflops==0.6.4
#tensorboardx
#tensorboard
#opencv-python
#torch-scope
#git+https://github.com/sovrasov/flops-counter.pytorch.git
#git+https://github.com/Tramac/torchscope.git

  • (required) Build and install apex to accelerate the training
    (see yuque),
    a little bit faster than pytorch DistributedDataParallel.

  • Put the imagenet data in ./data Using the following script:

cd TADL_DIR/pytorch/cream/
ln -s /mnt/data .

0x02 Quick Start

  • Run the following script to search an architecture.
python trainer.py
  • Selector (deprecated)
python selector.py
  • Train searched architectures.

Note: exponential moving average(model_ema) is not available yet.

python retrainer.py
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

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

Java Vue Python Text JavaScript other