Skip to content

Commit 12f2b9c

Browse files
darrengarveyacmiyaguchi
authored andcommitted
cifar10: Fix examples by setting snapshot_format.
Commit 4227828 set the default binary format from HDF5 to BINARYPROTO to fix BVLC#2885. This broke the cifar10 examples which relied on this default. This commit specifies the snapshot_format explicitly since the rest of the example relies on this being HDF5.
1 parent 0f5cfa2 commit 12f2b9c

5 files changed

+5
-0
lines changed

examples/cifar10/cifar10_full_solver.prototxt

+1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ display: 200
2121
max_iter: 60000
2222
# snapshot intermediate results
2323
snapshot: 10000
24+
snapshot_format: HDF5
2425
snapshot_prefix: "examples/cifar10/cifar10_full"
2526
# solver mode: CPU or GPU
2627
solver_mode: GPU

examples/cifar10/cifar10_full_solver_lr1.prototxt

+1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ display: 200
2121
max_iter: 65000
2222
# snapshot intermediate results
2323
snapshot: 5000
24+
snapshot_format: HDF5
2425
snapshot_prefix: "examples/cifar10/cifar10_full"
2526
# solver mode: CPU or GPU
2627
solver_mode: GPU

examples/cifar10/cifar10_full_solver_lr2.prototxt

+1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ display: 200
2121
max_iter: 70000
2222
# snapshot intermediate results
2323
snapshot: 5000
24+
snapshot_format: HDF5
2425
snapshot_prefix: "examples/cifar10/cifar10_full"
2526
# solver mode: CPU or GPU
2627
solver_mode: GPU

examples/cifar10/cifar10_quick_solver.prototxt

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ display: 100
2020
max_iter: 4000
2121
# snapshot intermediate results
2222
snapshot: 4000
23+
snapshot_format: HDF5
2324
snapshot_prefix: "examples/cifar10/cifar10_quick"
2425
# solver mode: CPU or GPU
2526
solver_mode: GPU

examples/cifar10/cifar10_quick_solver_lr1.prototxt

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ display: 100
2020
max_iter: 5000
2121
# snapshot intermediate results
2222
snapshot: 5000
23+
snapshot_format: HDF5
2324
snapshot_prefix: "examples/cifar10/cifar10_quick"
2425
# solver mode: CPU or GPU
2526
solver_mode: GPU

0 commit comments

Comments
 (0)