diff options
author | Fisher Yu <i@yf.io> | 2015-12-28 22:46:49 -0500 |
---|---|---|
committer | Fisher Yu <i@yf.io> | 2015-12-28 22:46:49 -0500 |
commit | 708c1a122c33bd35b0d53630fb74965488e1947a (patch) | |
tree | 6ce098c895c5804e774b8f1a7eb2293f0b52622f | |
parent | 08c5dfd53e6fd98148d6ce21e590407e38055984 (diff) | |
download | caffeonacl-708c1a122c33bd35b0d53630fb74965488e1947a.tar.gz caffeonacl-708c1a122c33bd35b0d53630fb74965488e1947a.tar.bz2 caffeonacl-708c1a122c33bd35b0d53630fb74965488e1947a.zip |
remove extra space before +
-rw-r--r-- | src/caffe/solvers/adam_solver.cpp | 2 | ||||
-rw-r--r-- | tools/caffe.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/src/caffe/solvers/adam_solver.cpp b/src/caffe/solvers/adam_solver.cpp index cb0fbfe2..c3378d38 100644 --- a/src/caffe/solvers/adam_solver.cpp +++ b/src/caffe/solvers/adam_solver.cpp @@ -30,7 +30,7 @@ void AdamSolver<Dtype>::ComputeUpdateValue(int param_id, Dtype rate) { Blob<Dtype>* val_v = this->history_[param_id + update_history_offset].get(); Blob<Dtype>* val_t = this->temp_[param_id].get(); - const int t = this->iter_ + 1; + const int t = this->iter_ + 1; const Dtype correction = std::sqrt(Dtype(1) - pow(beta2, t)) / (Dtype(1.) - pow(beta1, t)); const int N = net_params[param_id]->count(); diff --git a/tools/caffe.cpp b/tools/caffe.cpp index 305cfc36..6b342ace 100644 --- a/tools/caffe.cpp +++ b/tools/caffe.cpp @@ -164,7 +164,7 @@ int train() { if (FLAGS_gpu.size() == 0 && solver_param.solver_mode() == caffe::SolverParameter_SolverMode_GPU) { if (solver_param.has_device_id()) { - FLAGS_gpu = "" + + FLAGS_gpu = "" + boost::lexical_cast<string>(solver_param.device_id()); } else { // Set default GPU if unspecified FLAGS_gpu = "" + boost::lexical_cast<string>(0); |