Add more algorithms

2019-09-28 18:24:47 +10:00
parent bfd6b648fd
commit cfb462e463
286 changed files with 10557 additions and 122955 deletions
--- a/scripts/KD-train.sh
+++ b/scripts/KD-train.sh
@@ -0,0 +1,50 @@
+#!/bin/bash
+# bash ./scripts/KD-train.sh cifar10 ResNet110 ResNet110 0.5 1 -1
+echo script name: $0
+echo $# arguments
+if [ "$#" -ne 6 ] ;then
+  echo "Input illegal number of parameters " $#
+  echo "Need 6 parameters for the dataset / the-model-name / the-teacher-path / KD-alpha / KD-temperature / the-random-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
+model=$2
+teacher=$3
+alpha=$4
+temperature=$5
+epoch=E300
+LR=L1
+batch=256
+rseed=$6
+
+save_dir=./output/KD/${dataset}-${teacher}.2.${model}-${alpha}-${temperature}
+rm -rf ${save_dir}
+
+PY_C="./env/bin/python"
+if [ ! -f ${PY_C} ]; then
+  echo "Local Run with Python: "`which python`
+  PY_C="python"
+else
+  echo "Cluster Run with Python: "${PY_C}
+fi
+
+${PY_C} --version
+
+${PY_C} ./exps/KD-main.py --dataset ${dataset} \
+	--data_path $TORCH_HOME/cifar.python \
+	--model_config  ./configs/archs/CIFAR-${model}.config \
+	--optim_config  ./configs/opts/CIFAR-${epoch}-W5-${LR}-COS.config \
+	--KD_checkpoint $TORCH_HOME/TAS-checkpoints/basemodels/${dataset}/${teacher}.pth \
+	--procedure    Simple-KD \
+	--save_dir     ${save_dir} \
+	--KD_alpha ${alpha} --KD_temperature ${temperature} \
+	--cutout_length -1 \
+	--batch_size  ${batch} --rand_seed ${rseed} --workers 4 \
+	--eval_frequency 1 --print_freq 100 --print_freq_eval 200
--- a/scripts/base-imagenet.sh
+++ b/scripts/base-imagenet.sh
@@ -0,0 +1,74 @@
+#!/usr/bin/env sh
+# bash ./scripts/base-imagenet.sh ResNet110 Step-Soft 256
+set -e
+if [ "$#" -ne 4 ] ;then
+  echo "Input illegal number of parameters " $#
+  echo "Need 4 parameters for the architecture, and softmax/smooth-softmax, and batch-size, 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
+
+arch=$1
+opt=$2
+batch=$3
+rand_seed=$4
+dataset=imagenet-1k
+
+PY_C="./env/bin/python"
+if [ ! -f ${PY_C} ]; then
+  echo "Local Run with Python: "`which python`
+  PY_C="python"
+  SAVE_ROOT="./output"
+else
+  echo "Cluster Run with Python: "${PY_C}
+  SAVE_ROOT="./hadoop-data/SearchCheckpoints"
+  echo "Unzip ILSVRC2012"
+  tar --version
+  tar -xf ./hadoop-data/ILSVRC2012.tar -C ${TORCH_HOME}
+  echo "Unzip ILSVRC2012 done"
+fi
+
+if [ ${opt} = "RMSProp" ]; then
+  epoch=E200
+elif [ ${opt} = "Shuffle" ]; then
+  epoch=E240
+  dataset=imagenet-1k
+elif [ ${opt} = "MobileS" ]; then
+  epoch=E480
+  dataset=imagenet-1k-s
+elif [ ${opt} = "MobileFast" ] || [ ${opt} = "MobileFastS" ]; then
+  epoch=E150
+  dataset=imagenet-1k-s
+else
+  epoch=E120
+fi
+
+if [ ${batch} = "256" ]; then
+  opt_dir=opts
+  workers=24
+elif [ ${batch} = "1024" ]; then
+  opt_dir=opts-1K
+  workers=48
+else
+  echo "Invalid batch size : "${batch}
+  exit 1
+fi
+
+save_dir=${SAVE_ROOT}/basic/${dataset}/${arch}-${opt}-${epoch}-${batch}
+
+${PY_C} --version
+
+${PY_C} ./exps/basic-main.py --dataset ${dataset} \
+	--data_path $TORCH_HOME/ILSVRC2012 \
+	--model_config ./configs/archs/ImageNet-${arch}.config \
+	--optim_config ./configs/${opt_dir}/ImageNet-${epoch}-${opt}.config \
+	--procedure    basic \
+	--save_dir     ${save_dir} \
+	--cutout_length -1 \
+	--batch_size  ${batch} --rand_seed ${rand_seed} --workers ${workers} \
+	--eval_frequency 1 --print_freq 500 --print_freq_eval 2000
--- a/scripts/base-train.sh
+++ b/scripts/base-train.sh
@@ -0,0 +1,47 @@
+#!/bin/bash
+# bash ./scripts/base-train.sh cifar10 ResNet110 E300 L1 256 -1
+echo script name: $0
+echo $# arguments
+if [ "$#" -ne 6 ] ;then
+  echo "Input illegal number of parameters " $#
+  echo "Need 6 parameters for the dataset and the-model-name and epochs and LR and the-batch-size and the-random-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
+model=$2
+epoch=$3
+LR=$4
+batch=$5
+rseed=$6
+
+
+PY_C="./env/bin/python"
+if [ ! -f ${PY_C} ]; then
+  echo "Local Run with Python: "`which python`
+  PY_C="python"
+  SAVE_ROOT="./output"
+else
+  echo "Cluster Run with Python: "${PY_C}
+  SAVE_ROOT="./hadoop-data/SearchCheckpoints"
+fi
+
+save_dir=${SAVE_ROOT}/basic/${dataset}/${model}-${epoch}-${LR}-${batch}
+
+${PY_C} --version
+
+${PY_C} ./exps/basic-main.py --dataset ${dataset} \
+	--data_path $TORCH_HOME/cifar.python \
+	--model_config ./configs/archs/CIFAR-${model}.config \
+	--optim_config ./configs/opts/CIFAR-${epoch}-W5-${LR}-COS.config \
+	--procedure    basic \
+	--save_dir     ${save_dir} \
+	--cutout_length -1 \
+	--batch_size  ${batch} --rand_seed ${rseed} --workers 4 \
+	--eval_frequency 1 --print_freq 100 --print_freq_eval 200
--- a/scripts/clean.sh
+++ b/scripts/clean.sh
@@ -0,0 +1,6 @@
+#!/bin/bash
+find -name "__pycache__" | xargs rm -rf
+find -name "._.DS_Store" | xargs rm -rf
+find -name ".DS_Store"   | xargs rm -rf
+rm -rf output
+rm -rf ./scripts-cluster/tmps/job*
--- a/scripts/com-paddle.sh
+++ b/scripts/com-paddle.sh
@@ -0,0 +1,69 @@
+#!/bin/bash
+# bash ./scripts/com-paddle.sh cifar10
+set -e
+echo script name: $0
+echo $# arguments
+if [ "$#" -ne 1 ] ;then
+  echo "Input illegal number of parameters " $#
+  echo "Need 1 parameters for the dataset and the-model-name and the-random-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
+
+PY_C="./env/bin/python"
+if [ ! -f ${PY_C} ]; then
+  echo "Local Run with Python: "`which python`
+  PY_C="python"
+  SAVE_ROOT="./output/com-paddle"
+else
+  echo "Cluster Run with Python: "${PY_C}
+  SAVE_ROOT="./hadoop-data/COM-PADDLE"
+fi
+
+basic_func(){
+  dataset=$1
+  model=$2
+  batch=$3
+  rseed=$4
+  save_dir=${SAVE_ROOT}/${dataset}-${model}-${batch}
+  ${PY_C}  ./exps/basic-main.py --dataset ${dataset} \
+	--data_path $TORCH_HOME/cifar.python \
+	--model_config ./configs/archs/CIFAR-${model}.config \
+	--optim_config ./configs/opts/Com-Paddle-RES.config \
+	--procedure    basic \
+	--save_dir ${save_dir} --cutout_length -1 --batch_size  ${batch} --rand_seed ${rseed} --workers 4 --eval_frequency 1 --print_freq 100 --print_freq_eval 200
+}
+
+nas_infer_func(){
+  dataset=$1
+  model=$2
+  batch=$3
+  rseed=$4
+  save_dir=${SAVE_ROOT}/${dataset}-${model}-${batch}
+  ${PY_C}  ./exps/basic-main.py --dataset ${dataset} \
+	--data_path $TORCH_HOME/cifar.python --model_source nas \
+	--model_config ./configs/archs/NAS-CIFAR-${model}.config \
+	--optim_config ./configs/opts/Com-Paddle-NAS.config \
+	--procedure    basic \
+	--save_dir ${save_dir} --cutout_length -1 --batch_size  ${batch} --rand_seed ${rseed} --workers 4 --eval_frequency 1 --print_freq 100 --print_freq_eval 200
+}
+
+#datasets="cifar10 cifar100"
+
+#datasets="cifar10 cifar100"
+#for dataset in ${datasets}
+#do
+#basic_func ${dataset} ResNet20  256 -1
+#basic_func ${dataset} ResNet32  256 -1
+#basic_func ${dataset} ResNet110 256 -1
+#done
+
+nas_infer_func ${dataset} GDAS_V1 96 -1
+nas_infer_func ${dataset} SETN    96 -1
--- a/scripts/nas-infer-train.sh
+++ b/scripts/nas-infer-train.sh
@@ -0,0 +1,50 @@
+#!/bin/bash
+# bash ./scripts/nas-infer-train.sh cifar10 SETN 256 -1
+echo script name: $0
+echo $# arguments
+if [ "$#" -ne 4 ] ;then
+  echo "Input illegal number of parameters " $#
+  echo "Need 4 parameters for dataset and the-model-name and epochs and LR and the-batch-size and the-random-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
+model=$2
+batch=$3
+rseed=$4
+
+if [ ${dataset} == 'cifar10' ] || [ ${dataset} == 'cifar100' ]; then
+  xpath=$TORCH_HOME/cifar.python
+  base=CIFAR
+  workers=4
+  cutout_length=16
+elif [ ${dataset} == 'imagenet-1k' ]; then
+  xpath=$TORCH_HOME/ILSVRC2012
+  base=IMAGENET
+  workers=28
+  cutout_length=-1
+else
+  echo 'Unknown dataset: '${dataset}
+fi
+
+SAVE_ROOT="./output"
+
+save_dir=${SAVE_ROOT}/nas-infer/${dataset}-${model}-${batch}
+
+python --version
+
+python ./exps/basic-main.py --dataset ${dataset} \
+	--data_path ${xpath} --model_source nas \
+	--model_config ./configs/archs/NAS-${base}-${model}.config \
+	--optim_config ./configs/opts/NAS-${base}.config \
+	--procedure    basic \
+	--save_dir     ${save_dir} \
+	--cutout_length ${cutout_length} \
+	--batch_size  ${batch} --rand_seed ${rseed} --workers ${workers} \
+	--eval_frequency 1 --print_freq 500 --print_freq_eval 1000
--- a/scripts/prepare.sh
+++ b/scripts/prepare.sh
@@ -0,0 +1,13 @@
+#!/bin/bash
+# bash ./scripts/prepare.sh
+datasets='cifar10 cifar100 imagenet-1k'
+#ratios='0.5 0.8 0.9'
+ratios='0.5'
+save_dir=./.latent-data/splits
+
+for ratio in ${ratios}
+do
+  python ./exps/prepare.py --name cifar10  --root $TORCH_HOME/cifar.python  --save ${save_dir}/cifar10-${ratio}.pth --ratio ${ratio}
+  python ./exps/prepare.py --name cifar100 --root $TORCH_HOME/cifar.python  --save ${save_dir}/cifar100-${ratio}.pth --ratio ${ratio}
+  python ./exps/prepare.py --name imagenet-1k --root $TORCH_HOME/ILSVRC2012 --save ${save_dir}/imagenet-1k-${ratio}.pth --ratio ${ratio}
+done