changlin31 / DNA

(CVPR 2020) Block-wisely Supervised Neural Architecture Search with Knowledge Distillation
235 stars 35 forks source link

Add trainning module #9

Closed lucienne999 closed 4 years ago

lucienne999 commented 4 years ago

The Training Module is simplified from the repo: pytorch-image-models

Compared to the referenced repo, I removed irrelevant files and models.

The training step as follow:

  1. cd trainer
  2. Modify the run_example.sh: change data path and hyper-params according to your requirements
  3. ./run_example.sh