Skip to content

Commit 1d8024c

Browse files
jeffdonahuecbfinn
authored andcommitted
temporarily switch the snapshot_format default back to BINARYPROTO
out of anticipation for user issues due to issue BVLC#2885, which causes Caffe to crash when it attempts to snapshot nets with duplicate layer names
1 parent a98c8b1 commit 1d8024c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/caffe/proto/caffe.proto

+1-1
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ message SolverParameter {
190190
HDF5 = 0;
191191
BINARYPROTO = 1;
192192
}
193-
optional SnapshotFormat snapshot_format = 37 [default = HDF5];
193+
optional SnapshotFormat snapshot_format = 37 [default = BINARYPROTO];
194194
// the mode solver will use: 0 for CPU and 1 for GPU. Use GPU in default.
195195
enum SolverMode {
196196
CPU = 0;

src/caffe/test/test_gradient_based_solver.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ class GradientBasedSolverTest : public MultiDeviceTest<TypeParam> {
192192
if (snapshot) {
193193
ostringstream resume_file;
194194
resume_file << snapshot_prefix_ << "/_iter_" << num_iters
195-
<< ".solverstate.h5";
195+
<< ".solverstate";
196196
string resume_filename = resume_file.str();
197197
return resume_filename;
198198
}

0 commit comments

Comments
 (0)