Commit 73058c8 1 parent 0b4364f commit 73058c8 Copy full SHA for 73058c8
File tree 4 files changed +6
-4
lines changed
4 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -61,7 +61,9 @@ Makefile.config
61
61
data /*
62
62
models /*
63
63
* .caffemodel
64
+ * .caffemodel.h5
64
65
* .solverstate
66
+ * .solverstate.h5
65
67
* .binaryproto
66
68
* leveldb
67
69
* lmdb
Original file line number Diff line number Diff line change @@ -8,9 +8,9 @@ $TOOLS/caffe train \
8
8
# reduce learning rate by factor of 10
9
9
$TOOLS /caffe train \
10
10
--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
12
12
13
13
# reduce learning rate by factor of 10
14
14
$TOOLS /caffe train \
15
15
--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
Original file line number Diff line number Diff line change @@ -8,4 +8,4 @@ $TOOLS/caffe train \
8
8
# reduce learning rate by factor of 10 after 8 epochs
9
9
$TOOLS /caffe train \
10
10
--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
Original file line number Diff line number Diff line change 2
2
3
3
./build/tools/caffe train \
4
4
--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
You can’t perform that action at this time.
0 commit comments