2019-01-31 15:27:38 +01:00
|
|
|
# GDAS
|
|
|
|
By Xuanyi Dong and Yi Yang
|
|
|
|
|
|
|
|
University of Technology Sydney
|
|
|
|
|
|
|
|
Requirements
|
|
|
|
- PyTorch 1.0
|
|
|
|
- Python 3.6
|
2019-01-31 18:03:35 +01:00
|
|
|
- opencv
|
2019-01-31 15:27:38 +01:00
|
|
|
```
|
|
|
|
conda install pytorch torchvision cuda100 -c pytorch
|
|
|
|
```
|
|
|
|
|
|
|
|
## Algorithm
|
|
|
|
|
|
|
|
Searching CNNs
|
|
|
|
```
|
2019-01-31 18:06:28 +01:00
|
|
|
bash ./scripts-cnn/search.sh 1 base cifar10
|
2019-02-01 07:17:02 +01:00
|
|
|
bash ./scripts-cnn/DMS-V-TrainV3.sh 1
|
2019-01-31 18:03:35 +01:00
|
|
|
bash ./scripts-cnn/search-acc-v2.sh 3 acc2
|
2019-01-31 15:27:38 +01:00
|
|
|
```
|
2019-01-31 16:40:20 +01:00
|
|
|
|
2019-01-31 17:23:55 +01:00
|
|
|
Train the searched CNN on CIFAR
|
|
|
|
```
|
2019-02-02 12:27:36 +01:00
|
|
|
bash ./scripts-cnn/train-cifar.sh 0 GDAS_FG cifar10 cut
|
|
|
|
bash ./scripts-cnn/train-cifar.sh 0 GDAS_F1 cifar10 cut
|
|
|
|
bash ./scripts-cnn/train-cifar.sh 0 GDAS_V1 cifar100 cut
|
2019-01-31 17:23:55 +01:00
|
|
|
```
|
|
|
|
|
|
|
|
Train the searched CNN on ImageNet
|
|
|
|
```
|
|
|
|
bash ./scripts-cnn/train-imagenet.sh 0 GDAS_F1 52 14
|
|
|
|
bash ./scripts-cnn/train-imagenet.sh 0 GDAS_V1 50 14
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
|
|
Train the searched RNN
|
2019-01-31 16:40:20 +01:00
|
|
|
```
|
|
|
|
bash ./scripts-rnn/train-PTB.sh 0 DARTS_V1
|
|
|
|
bash ./scripts-rnn/train-PTB.sh 0 DARTS_V2
|
|
|
|
bash ./scripts-rnn/train-PTB.sh 0 GDAS
|
|
|
|
bash ./scripts-rnn/train-WT2.sh 0 DARTS_V1
|
|
|
|
bash ./scripts-rnn/train-WT2.sh 0 DARTS_V2
|
|
|
|
bash ./scripts-rnn/train-WT2.sh 0 GDAS
|
|
|
|
```
|