Skip to content

Commit 73058c8

Browse files
committed
Update example bash scripts to expect .h5, new extensions in .gitignore
1 parent 0b4364f commit 73058c8

File tree

4 files changed

+6
-4
lines changed

4 files changed

+6
-4
lines changed

.gitignore

+2
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,9 @@ Makefile.config
6161
data/*
6262
models/*
6363
*.caffemodel
64+
*.caffemodel.h5
6465
*.solverstate
66+
*.solverstate.h5
6567
*.binaryproto
6668
*leveldb
6769
*lmdb

examples/cifar10/train_full.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@ $TOOLS/caffe train \
88
# reduce learning rate by factor of 10
99
$TOOLS/caffe train \
1010
--solver=examples/cifar10/cifar10_full_solver_lr1.prototxt \
11-
--snapshot=examples/cifar10/cifar10_full_iter_60000.solverstate
11+
--snapshot=examples/cifar10/cifar10_full_iter_60000.solverstate.h5
1212

1313
# reduce learning rate by factor of 10
1414
$TOOLS/caffe train \
1515
--solver=examples/cifar10/cifar10_full_solver_lr2.prototxt \
16-
--snapshot=examples/cifar10/cifar10_full_iter_65000.solverstate
16+
--snapshot=examples/cifar10/cifar10_full_iter_65000.solverstate.h5

examples/cifar10/train_quick.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,4 @@ $TOOLS/caffe train \
88
# reduce learning rate by factor of 10 after 8 epochs
99
$TOOLS/caffe train \
1010
--solver=examples/cifar10/cifar10_quick_solver_lr1.prototxt \
11-
--snapshot=examples/cifar10/cifar10_quick_iter_4000.solverstate
11+
--snapshot=examples/cifar10/cifar10_quick_iter_4000.solverstate.h5

examples/imagenet/resume_training.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22

33
./build/tools/caffe train \
44
--solver=models/bvlc_reference_caffenet/solver.prototxt \
5-
--snapshot=models/bvlc_reference_caffenet/caffenet_train_10000.solverstate
5+
--snapshot=models/bvlc_reference_caffenet/caffenet_train_10000.solverstate.h5

0 commit comments

Comments
 (0)