NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Do the bias addition in the convolution in a separate call (#2656)

Browse Source 
* Do the bias addition in the convolution in a separate call

Fixes #2655.
Somehow, doing it in a single call is slower.

* Grammar

* The change should be this, really
This commit is contained in:
Adrià Arrufat 2022-09-01 20:39:19 +09:00 committed by GitHub
parent da2f45b2e5
commit 902c70193a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
dlib/dnn/layers.h
View File

@ -232,19 +232,18 @@ namespace dlib
_stride_x, _stride_x,
padding_y_, padding_y_,
padding_x_); padding_x_);
conv(false, output,
sub.get_output(),
filters(params,0));
// For some reason, doing this is sometimes slower than two separate calls
// conv(false, output,
// sub.get_output(),
// filters(params,0),
// biases(params, filters.size()));
if (use_bias) if (use_bias)
{ tt::add(1,output,1,biases(params,filters.size()));
conv(false, output,
sub.get_output(),
filters(params,0),
biases(params, filters.size()));
}
else
{
conv(false, output,
sub.get_output(),
filters(params,0));
}
} }
template <typename SUBNET> template <typename SUBNET>