Add experiment scripts (#10)

* Add experiments

Author: MattPainter01

experiments/cifar_experiment.sh

@@ -0,0 +1,53 @@
+
+# Usage: `bash imagenet_experiment (cifar) (model) (msda_mode) (dataset path)`
+# where:
+# cifar is one of [cifar10, cifar100]
+# model is one of [resnet, wrn, densenet, pyramidnet]
+# msda_mode is one of [fmix, mixup, None]
+# For multiple GPU, add --parallel=True
+
+if [ "$1" == "cifar10" ]
+then
+    ds=cifar10
+fi
+if [ "$1" == "cifar100" ]
+then
+    ds=cifar10
+fi
+
+if [ "$2" == "resnet" ]
+then
+    model=ResNet18
+    epoch=200
+    schedule=(100 150)
+    bs=128
+    cosine=False
+fi
+
+if [ "$2" == "wrn" ]
+then
+    model=wrn
+    epoch=200
+    schedule=(100 150)
+    bs=128
+    cosine=False
+fi
+
+if [ "$2" == "densenet" ]
+then
+    model=DenseNet190
+    epoch=300
+    schedule=(100 150 225)
+    bs=32
+    cosine=False
+fi
+
+if [ "$2" == "pyramidnet" ]
+then
+    model=aa_PyramidNet
+    epoch=1800
+    schedule=2000
+    cosine=True
+fi
+
+python ../trainer.py --dataset=$ds --model=$model --epoch=$epoch --schedule=$schedule --lr=0.1 --dataset-path=$4 --msda-mode=$3 --batch-size=$bs --cosine-scheduler=$cosine

experiments/fashion_experiments.sh

@@ -0,0 +1,32 @@
+
+# Usage: `bash imagenet_experiment (model) (msda_mode) (dataset path)`
+# where:
+# model is one of [resnet, wrn, densenet]
+# msda_mode is one of [fmix, mixup, None]
+# For multiple GPU, add --parallel=True
+
+if [ "$1" == "resnet" ]
+then
+    model=ResNet18
+    epoch=200
+    schedule=(100 150)
+    bs=128
+fi
+
+if [ "$1" == "wrn" ]
+then
+    model=wrn
+    epoch=300
+    schedule=(100 150 225)
+    bs=32
+fi
+
+if [ "$1" == "densenet" ]
+then
+    model=DenseNet190
+    epoch=300
+    schedule=(100 150 225)
+    bs=32
+fi
+
+python ../trainer.py --dataset=fashion --model=$model --epoch=$epoch --schedule=$schedule --lr=0.1 --dataset-path=$3 --msda-mode=$2 --batch-size=$bs

experiments/google_commands_experiment.sh

@@ -0,0 +1,6 @@
+
+# Usage: `bash imagenet_experiment (msda_mode) (dataset path) (alpha)`
+# where msda_mode is one of [fmix, mixup, None]
+# For multiple GPU, add --parallel=True
+
+python ../trainer.py --dataset=commands --epoch=90 --schedule=30 60 80 --lr=0.01 --dataset-path=$2 --msda-mode=$1 --alpha=$3

experiments/imagenet_experiment.sh

@@ -0,0 +1,7 @@
+
+
+# Usage: `bash imagenet_experiment (msda_mode) (dataset path)`
+# where msda_mode is one of [fmix, mixup, None]
+# For multiple GPU, add --parallel=True
+
+python ../trainer.py --dataset=imagenet --epoch=90 --model=torch_resnet101 --schedule 30 60 80 --batch-size=256 --lr=0.4 --lr-warmup=True --dataset-path=$2 --msda-mode=$1

experiments/modelnet_experiment.sh

@@ -0,0 +1,6 @@
+
+# Usage: `bash imagenet_experiment (msda_mode) (dataset path)`
+# where msda_mode is one of [fmix, mixup, None]
+# For multiple GPU, add --parallel=True
+
+python ../trainer.py --dataset=modelnet --epoch=50 --schedule=10 20 30 40 --lr=0.001 --dataset-path=$2 --msda-mode=$1 --batch-size=16

experiments/tiny_imagenet_experiment.sh

@@ -0,0 +1,6 @@
+
+# Usage: `bash imagenet_experiment (msda_mode) (dataset path)`
+# where msda_mode is one of [fmix, mixup, None]
+# For multiple GPU, add --parallel=True
+
+python ../trainer.py --dataset=tinyimagenet --epoch=200 --schedule 150 180 --batch-size=128 --lr=0.1 --dataset-path=$2 --msda-mode=$1

experiments/toxic_experiment.sh

@@ -0,0 +1,6 @@
+
+# Usage: `bash imagenet_experiment (msda_mode) (dataset path)`
+# where msda_mode is one of [fmix, mixup, None]
+# For multiple GPU, add --parallel=True
+
+python ../trainer.py --dataset=toxic --epoch=10 --batch-size=64 --lr=1e-4 --dataset-path=$2 --msda-mode=$1