Skip to content

Commit e5991dc

Browse files
Fix/bash schedule (#27)
* Fix schedule * Remove unnecessary change * Update experiments/cifar_experiment.sh * Fix modelnet schedule Co-authored-by: Ethan Harris <[email protected]> Co-authored-by: Ethan Harris <[email protected]>
1 parent 615f77c commit e5991dc

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

experiments/cifar_experiment.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -51,4 +51,4 @@ then
5151
cosine=True
5252
fi
5353

54-
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
54+
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

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,4 +29,4 @@ then
2929
bs=32
3030
fi
3131

32-
python ../trainer.py --dataset=fashion --model=$model --epoch=$epoch --schedule=$schedule --lr=0.1 --dataset-path=$3 --msda-mode=$2 --batch-size=$bs
32+
python ../trainer.py --dataset=fashion --model=$model --epoch=$epoch --schedule ${schedule[@]} --lr=0.1 --dataset-path=$3 --msda-mode=$2 --batch-size=$bs

experiments/modelnet_experiment.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
# where msda_mode is one of [fmix, mixup, None]
44
# For multiple GPU, add --parallel=True
55

6-
python ../trainer.py --dataset=modelnet --epoch=50 --schedule=10 20 30 40 --lr=0.001 --dataset-path=$2 --msda-mode=$1 --batch-size=16
6+
python ../trainer.py --dataset=modelnet --epoch=50 --schedule 10 20 30 40 --lr=0.001 --dataset-path=$2 --msda-mode=$1 --batch-size=16

0 commit comments

Comments
 (0)