#!/bin/bash # bash ./scripts-search/train-models.sh 0/1 0 100 -1 '777 888 999' echo script name: $0 echo $# arguments if [ "$#" -ne 5 ] ;then echo "Input illegal number of parameters " $# echo "Need 5 parameters for use-less-or-not, start-and-end, arch-index, and seeds" 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 use_less=$1 xstart=$2 xend=$3 arch_index=$4 all_seeds=$5 save_dir=./output/NAS-BENCH-201-4/ if [ ${arch_index} == "-1" ]; then mode=new else mode=cover fi # OMP_NUM_THREADS=4 python ./exps/NAS-Bench-201/main.py \ # --mode ${mode} --save_dir ${save_dir} --max_node 4 \ # --use_less ${use_less} \ # --datasets cifar10 cifar10 cifar100 ImageNet16-120 \ # --splits 1 0 0 0 \ # --xpaths $TORCH_HOME/cifar.python \ # $TORCH_HOME/cifar.python \ # $TORCH_HOME/cifar.python \ # $TORCH_HOME/cifar.python/ImageNet16 \ # --channel 16 --num_cells 5 \ # --workers 4 \ # --srange ${xstart} ${xend} --arch_index ${arch_index} \ # --seeds ${all_seeds} OMP_NUM_THREADS=4 python ./exps/NAS-Bench-201/main.py \ --mode ${mode} --save_dir ${save_dir} --max_node 4 \ --use_less ${use_less} \ --datasets aircraft \ --xpaths /lustre/hpe/ws11/ws11.1/ws/xmuhanma-SWAP/train_datasets/datasets/ \ --channel 16 \ --splits 1 \ --num_cells 5 \ --workers 4 \ --srange ${xstart} ${xend} --arch_index ${arch_index} \ --seeds ${all_seeds} # OMP_NUM_THREADS=4 python ./exps/NAS-Bench-201/main.py \ # --mode ${mode} --save_dir ${save_dir} --max_node 4 \ # --use_less ${use_less} \ # --datasets oxford\ # --xpaths /lustre/hpe/ws11/ws11.1/ws/xmuhanma-SWAP/train_datasets/datasets/ \ # --channel 16 \ # --splits 1 \ # --num_cells 5 \ # --workers 4 \ # --srange ${xstart} ${xend} --arch_index ${arch_index} \ # --seeds ${all_seeds}