xautodl/README.md
2019-02-01 02:40:20 +11:00

489 B

GDAS

By Xuanyi Dong and Yi Yang

University of Technology Sydney

Requirements

  • PyTorch 1.0
  • Python 3.6
conda install pytorch torchvision cuda100 -c pytorch

Algorithm

Searching CNNs

Train the Searched RNN

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