Skip to content

Commit

Permalink
Update example bash scripts to expect .h5, new extensions in .gitignore
Browse files Browse the repository at this point in the history
  • Loading branch information
erictzeng committed Aug 7, 2015
1 parent 5c89c64 commit c9b333e
Showing 4 changed files with 6 additions and 4 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -61,7 +61,9 @@ Makefile.config
data/*
models/*
*.caffemodel
*.caffemodel.h5
*.solverstate
*.solverstate.h5
*.binaryproto
*leveldb
*lmdb
4 changes: 2 additions & 2 deletions examples/cifar10/train_full.sh
Original file line number Diff line number Diff line change
@@ -8,9 +8,9 @@ $TOOLS/caffe train \
# reduce learning rate by factor of 10
$TOOLS/caffe train \
--solver=examples/cifar10/cifar10_full_solver_lr1.prototxt \
--snapshot=examples/cifar10/cifar10_full_iter_60000.solverstate
--snapshot=examples/cifar10/cifar10_full_iter_60000.solverstate.h5

# reduce learning rate by factor of 10
$TOOLS/caffe train \
--solver=examples/cifar10/cifar10_full_solver_lr2.prototxt \
--snapshot=examples/cifar10/cifar10_full_iter_65000.solverstate
--snapshot=examples/cifar10/cifar10_full_iter_65000.solverstate.h5
2 changes: 1 addition & 1 deletion examples/cifar10/train_quick.sh
Original file line number Diff line number Diff line change
@@ -8,4 +8,4 @@ $TOOLS/caffe train \
# reduce learning rate by factor of 10 after 8 epochs
$TOOLS/caffe train \
--solver=examples/cifar10/cifar10_quick_solver_lr1.prototxt \
--snapshot=examples/cifar10/cifar10_quick_iter_4000.solverstate
--snapshot=examples/cifar10/cifar10_quick_iter_4000.solverstate.h5
2 changes: 1 addition & 1 deletion examples/imagenet/resume_training.sh
Original file line number Diff line number Diff line change
@@ -2,4 +2,4 @@

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

0 comments on commit c9b333e

Please sign in to comment.