diff options
| author | Yan Chen <yanchen036@gmail.com> | 2015-09-18 17:02:16 +0800 |
|---|---|---|
| committer | Yan Chen <yanchen036@gmail.com> | 2015-09-18 17:02:16 +0800 |
| commit | f75d594bbec1efab69cdc09c04bed1762aebd0e1 (patch) | |
| tree | 2b3e5e575933122ff941df64218da5d82c7a795f /src | |
| parent | 3d12b5d9cb014d2e0df4db1b10250ef298e549b9 (diff) | |
| download | caffeonacl-f75d594bbec1efab69cdc09c04bed1762aebd0e1.tar.gz caffeonacl-f75d594bbec1efab69cdc09c04bed1762aebd0e1.tar.bz2 caffeonacl-f75d594bbec1efab69cdc09c04bed1762aebd0e1.zip | |
refine format of switch case 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); |