mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Test for XNOR-conv on CUDA

Browse Source
This commit is contained in:
AlexeyAB
2018-09-07 23:04:46 +03:00
parent 57e878b4f9
commit c0e01fd63c
5 changed files with 378 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/convolutional_layer.c
View File

@ -605,6 +605,7 @@ void binary_align_weights(convolutional_layer *l)
int m = l->n;
int k = l->size*l->size*l->c;
size_t new_lda = k + (l->lda_align - k % l->lda_align); // (k / 8 + 1) * 8;
l->new_lda = new_lda;
binarize_weights(l->weights, m, k, l->binary_weights);