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

Added fast_binarize_weights_gpu()

Browse Source
This commit is contained in:
AlexeyAB
2018-11-05 22:38:35 +03:00
parent 2c5e383c04
commit 25f65f6878
4 changed files with 99 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/network.c
View File

@ -866,10 +866,10 @@ void calculate_binary_weights(network net)
//if (l->size*l->size*l->c >= 2048) l->lda_align = 512;
binary_align_weights(l);
}
if (net.layers[j].use_bin_output) {
l->activation = LINEAR;
if (net.layers[j].use_bin_output) {
l->activation = LINEAR;
}
}
}
}