diff options
| author | Jon Long <jonlong@cs.berkeley.edu> | 2015-09-18 19:56:20 -0700 |
|---|---|---|
| committer | Jon Long <jonlong@cs.berkeley.edu> | 2015-09-18 19:56:20 -0700 |
| commit | 68655b55925cfa5cd6543cdfb879bf6e68bd2a3c (patch) | |
| tree | 92322a694aa32ff14deecc11e7aed2206219b3ff /src | |
| parent | 2e1c1cb7788eff438dba6ea77e100e1b3af51104 (diff) | |
| parent | f75d594bbec1efab69cdc09c04bed1762aebd0e1 (diff) | |
| download | caffeonacl-68655b55925cfa5cd6543cdfb879bf6e68bd2a3c.tar.gz caffeonacl-68655b55925cfa5cd6543cdfb879bf6e68bd2a3c.tar.bz2 caffeonacl-68655b55925cfa5cd6543cdfb879bf6e68bd2a3c.zip | |
Merge pull request #3083 from yanchen036/master
refine format of switch case statement in solver
Diffstat (limited to 'src')
| -rw-r--r-- | src/caffe/solver.cpp | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp index 3574ce75..12c13dd8 100644 --- a/src/caffe/solver.cpp +++ b/src/caffe/solver.cpp @@ -422,14 +422,14 @@ void Solver<Dtype>::Snapshot() { CHECK(Caffe::root_solver()); string model_filename; switch (param_.snapshot_format()) { - case caffe::SolverParameter_SnapshotFormat_BINARYPROTO: - model_filename = SnapshotToBinaryProto(); - break; - case caffe::SolverParameter_SnapshotFormat_HDF5: - model_filename = SnapshotToHDF5(); - break; - default: - LOG(FATAL) << "Unsupported snapshot format."; + case caffe::SolverParameter_SnapshotFormat_BINARYPROTO: + model_filename = SnapshotToBinaryProto(); + break; + case caffe::SolverParameter_SnapshotFormat_HDF5: + model_filename = SnapshotToHDF5(); + break; + default: + LOG(FATAL) << "Unsupported snapshot format."; } SnapshotSolverState(model_filename); |