Skip to content

Commit

Permalink
cifar10: Fix examples by setting snapshot_format.
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
darrengarvey authored and wangyida committed Sep 15, 2015
1 parent bed2133 commit 23f8699
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 0 deletions.
1 change: 1 addition & 0 deletions examples/cifar10/cifar10_full_solver.prototxt
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ display: 200
max_iter: 60000
# snapshot intermediate results
snapshot: 10000
snapshot_format: HDF5
snapshot_prefix: "examples/cifar10/cifar10_full"
# solver mode: CPU or GPU
solver_mode: GPU
1 change: 1 addition & 0 deletions examples/cifar10/cifar10_full_solver_lr1.prototxt
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ display: 200
max_iter: 65000
# snapshot intermediate results
snapshot: 5000
snapshot_format: HDF5
snapshot_prefix: "examples/cifar10/cifar10_full"
# solver mode: CPU or GPU
solver_mode: GPU
1 change: 1 addition & 0 deletions examples/cifar10/cifar10_full_solver_lr2.prototxt
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ display: 200
max_iter: 70000
# snapshot intermediate results
snapshot: 5000
snapshot_format: HDF5
snapshot_prefix: "examples/cifar10/cifar10_full"
# solver mode: CPU or GPU
solver_mode: GPU
1 change: 1 addition & 0 deletions examples/cifar10/cifar10_quick_solver.prototxt
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ display: 100
max_iter: 4000
# snapshot intermediate results
snapshot: 4000
snapshot_format: HDF5
snapshot_prefix: "examples/cifar10/cifar10_quick"
# solver mode: CPU or GPU
solver_mode: GPU
1 change: 1 addition & 0 deletions examples/cifar10/cifar10_quick_solver_lr1.prototxt
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ display: 100
max_iter: 5000
# snapshot intermediate results
snapshot: 5000
snapshot_format: HDF5
snapshot_prefix: "examples/cifar10/cifar10_quick"
# solver mode: CPU or GPU
solver_mode: GPU

0 comments on commit 23f8699

Please sign in to comment.