Skip to content
This repository was archived by the owner on Jan 8, 2021. It is now read-only.

Commit 184d7ae

Browse files
committed
Update network name
1 parent bd2d902 commit 184d7ae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

run.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@ train_mlp_mnist() {
66
python Train.py --epoch 20 --dataset mnist --network mlp --loss_func "$2" --model_name "$3" 784 "$1"
77
}
88
train_cnn_mnist() {
9-
python Train.py --epoch 20 --dataset mnist --network cnn --loss_func "$2" --model_name "$3" 1 "$1"
9+
python Train.py --epoch 20 --dataset mnist --network conv --loss_func "$2" --model_name "$3" 1 "$1"
1010
}
1111
train_mlp_cifar() {
1212
python Train.py --epoch 20 --dataset cifar10 --network mlp --loss_func "$2" --model_name "$3" 3072 "$1"
1313
}
1414
train_cnn_cifar() {
15-
python Train.py --epoch 20 --dataset cifar10 --network cnn --loss_func "$2" --model_name "$3" 3 "$1"
15+
python Train.py --epoch 20 --dataset cifar10 --network conv --loss_func "$2" --model_name "$3" 3 "$1"
1616
}

0 commit comments

Comments
 (0)