Skip to content

Commit 4b94e1d

Browse files
committed
Updating execution examples
1 parent 7936aad commit 4b94e1d

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

README.md

+7-5
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,16 @@ in order to reduce the model parameters.
3030
As an example, use the following command to train an MSDNet on Cifar10
3131

3232
```
33-
python3 main.py --model msdnet -b 64 -j 2 cifar10 --msd-blocks 10 --msd-base 2
33+
python3 main.py --model msdnet -b 64 -j 2 cifar10 --msd-blocks 10 --msd-base 4 \
3434
--msd-step 2 --msd-stepmode even --growth 6-12-24 --gpu 0
3535
```
3636

3737
As an example, use the following command to train an MSDNet on Cifar100 with GCN block
3838

3939
```
40-
python3 main.py --model msdnet -b 64 -j 2 cifar100 --msd-blocks 10 --msd-base 3
41-
--msd-step 2 --msd-stepmode even --growth 6-12-24 --gpu 0 --msd-gcn --msd-gcn-kernel 5
40+
python3 main.py --model msdnet -b 64 -j 2 cifar100 --msd-blocks 10 --msd-base 3 \
41+
--msd-step 1 --msd-stepmode even --growth 6-12-24 --gpu 0 --msd-gcn --msd-gcn-kernel 5 \
42+
--msd-share-weights --msd-all-gcn
4243
```
4344

4445

@@ -48,8 +49,9 @@ We take the Cifar10 model trained above as an example.
4849
To evaluate the trained model, use `evaluate` to evaluate from the default checkpoint directory:
4950

5051
```
51-
python main.py ---model msdnet -b 64 -j 2 cifar10 --msd-blocks 10 --msd-base 2
52-
--msd-step 2 --msd-stepmode even --growth 6-12-24 --gpu 0 --resume --evaluate
52+
python3 main.py --model msdnet -b 64 -j 2 cifar100 --msd-blocks 10 --msd-base 3 \
53+
--msd-step 1 --msd-stepmode even --growth 6-12-24 --gpu 0 --msd-gcn --msd-gcn-kernel 5 \
54+
--msd-share-weights --msd-all-gcn --resume --evaluate
5355
```
5456

5557

script.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -63,5 +63,5 @@ python3 main.py --model msdnet -b 64 -j 2 cifar10 --msd-blocks 10 --msd-base 2
6363

6464
# cifar100, all shared gcn, kernel 5, base 3, step 1
6565
python3 main.py --model msdnet -b 64 -j 2 cifar100 --msd-blocks 10 --msd-base 3 \
66-
--msd-step 1 --msd-stepmode even --growth 6-12-24 --gpu 1 --msd-gcn --msd-gcn-kernel 5 \
67-
--msd-share-weights --msd-all-gcn
66+
--msd-step 1 --msd-stepmode even --growth 6-12-24 --gpu 0 --msd-gcn --msd-gcn-kernel 5 \
67+
--msd-share-weights --msd-all-gcn

0 commit comments

Comments
 (0)