| 912 | |
| 913 | template <typename Dtype> |
| 914 | void Net<Dtype>::ClearParamDiffs() { |
| 915 | for (int i = 0; i < learnable_params_.size(); ++i) { |
| 916 | Blob<Dtype>* blob = learnable_params_[i]; |
| 917 | switch (Caffe::mode()) { |
| 918 | case Caffe::CPU: |
| 919 | caffe_set(blob->count(), static_cast<Dtype>(0), |
| 920 | blob->mutable_cpu_diff()); |
| 921 | break; |
| 922 | case Caffe::GPU: |
| 923 | #ifndef CPU_ONLY |
| 924 | caffe_gpu_set(blob->count(), static_cast<Dtype>(0), |
| 925 | blob->mutable_gpu_diff()); |
| 926 | #else |
| 927 | NO_GPU; |
| 928 | #endif |
| 929 | break; |
| 930 | } |
| 931 | } |
| 932 | } |
| 933 | |
| 934 | template <typename Dtype> |
| 935 | void Net<Dtype>::ShareWeights() { |
no test coverage detected