From: qipeng Date: Wed, 23 Jul 2014 17:25:44 +0000 (-0700) Subject: bugfixes for AdaGrad X-Git-Tag: submit/tizen/20180823.020014~620^2~62^2~17 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=76ef2ca1f6e326622090aa7d57a10e80d5831350;p=platform%2Fupstream%2Fcaffeonacl.git bugfixes for AdaGrad --- diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp index abcbe5e..8928c7b 100644 --- a/src/caffe/solver.cpp +++ b/src/caffe/solver.cpp @@ -577,14 +577,14 @@ void AdaGradSolver::ComputeUpdateValue() { if (local_decay) { // add weight decay caffe_axpy(net_params[param_id]->count(), - local_decay * local_rate, + local_decay, net_params[param_id]->cpu_data(), - this->history_[param_id]->mutable_cpu_data()); + net_params[param_id]->mutable_cpu_diff()); } // compute square of gradient in update caffe_powx(net_params[param_id]->count(), - net_params[param_id]->cpu_data(), Dtype(2), + net_params[param_id]->cpu_diff(), Dtype(2), this->update_[param_id]->mutable_cpu_data()); // update history @@ -621,14 +621,14 @@ void AdaGradSolver::ComputeUpdateValue() { if (local_decay) { // add weight decay caffe_gpu_axpy(net_params[param_id]->count(), - local_decay * local_rate, + local_decay, net_params[param_id]->gpu_data(), - this->history_[param_id]->mutable_gpu_data()); + net_params[param_id]->mutable_gpu_diff()); } // compute square of gradient in update caffe_gpu_powx(net_params[param_id]->count(), - net_params[param_id]->gpu_data(), Dtype(2), + net_params[param_id]->gpu_diff(), Dtype(2), this->update_[param_id]->mutable_gpu_data()); // update history