#!/bin/bash # Efficient Neural Architecture Search via Parameter Sharing, ICML 2018 # bash ./scripts-search/scripts/algos/ENAS.sh cifar10 -1 echo script name: $0 echo $# arguments if [ "$#" -ne 2 ] ;then echo "Input illegal number of parameters " $# echo "Need 2 parameters for dataset and seed" exit 1 fi if [ "$TORCH_HOME" = "" ]; then echo "Must set TORCH_HOME envoriment variable for data dir saving" exit 1 else echo "TORCH_HOME : $TORCH_HOME" fi dataset=$1 seed=$2 channel=16 num_cells=5 max_nodes=4 if [ "$dataset" == "cifar10" ] || [ "$dataset" == "cifar100" ]; then data_path="$TORCH_HOME/cifar.python" else data_path="$TORCH_HOME/cifar.python/ImageNet16" fi save_dir=./output/cell-search-tiny/ENAS-${dataset} OMP_NUM_THREADS=4 python ./exps/algos/ENAS.py \ --save_dir ${save_dir} --max_nodes ${max_nodes} --channel ${channel} --num_cells ${num_cells} \ --dataset ${dataset} --data_path ${data_path} \ --search_space_name aa-nas \ --config_path ./configs/nas-benchmark/algos/ENAS.config \ --controller_entropy_weight 0.0001 \ --controller_bl_dec 0.99 \ --controller_train_steps 50 \ --controller_num_aggregate 20 \ --controller_num_samples 100 \ --workers 4 --print_freq 200 --rand_seed ${seed}