зеркало из https://github.com/microsoft/caffe.git
Merge pull request #3605 from jeffdonahue/bias-gemm-type-fix
Remove incorrect cast of gemm int arg to Dtype in BiasLayer
This commit is contained in:
Коммит
0acb338b9a
|
@ -80,7 +80,7 @@ void BiasLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
|
||||||
}
|
}
|
||||||
for (int n = 0; n < outer_dim_; ++n) {
|
for (int n = 0; n < outer_dim_; ++n) {
|
||||||
caffe_cpu_gemm(CblasNoTrans, CblasNoTrans, bias_dim_,
|
caffe_cpu_gemm(CblasNoTrans, CblasNoTrans, bias_dim_,
|
||||||
inner_dim_, Dtype(1), Dtype(1), bias_data,
|
inner_dim_, 1, Dtype(1), bias_data,
|
||||||
bias_multiplier_.cpu_data(), Dtype(1), top_data);
|
bias_multiplier_.cpu_data(), Dtype(1), top_data);
|
||||||
top_data += dim_;
|
top_data += dim_;
|
||||||
}
|
}
|
||||||
|
|
Загрузка…
Ссылка в новой задаче