diff --git a/.gitignore b/.gitignore index 28f2aca854b..53c1fb056bb 100644 --- a/.gitignore +++ b/.gitignore @@ -61,7 +61,9 @@ Makefile.config data/* models/* *.caffemodel +*.caffemodel.h5 *.solverstate +*.solverstate.h5 *.binaryproto *leveldb *lmdb diff --git a/examples/cifar10/train_full.sh b/examples/cifar10/train_full.sh index 4285a5d6468..ef112e1f6db 100755 --- a/examples/cifar10/train_full.sh +++ b/examples/cifar10/train_full.sh @@ -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 diff --git a/examples/cifar10/train_quick.sh b/examples/cifar10/train_quick.sh index 2830c40945c..6b7d228879b 100755 --- a/examples/cifar10/train_quick.sh +++ b/examples/cifar10/train_quick.sh @@ -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 diff --git a/examples/imagenet/resume_training.sh b/examples/imagenet/resume_training.sh index d1febff8d39..bf7945c0fd0 100755 --- a/examples/imagenet/resume_training.sh +++ b/examples/imagenet/resume_training.sh @@ -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