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

XNOR-net tiny-yolo_xnor.cfg ~2x faster than cuDNN on CUDA (nVidia GPU Maxwell)

Browse Source
This commit is contained in:
AlexeyAB
2018-09-22 02:01:14 +03:00
parent 0224ba3d0d
commit 7dd97537fb
15 changed files with 511 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/network.c
View File

@ -793,6 +793,7 @@ void free_network(network net)
#ifdef GPU
if (gpu_index >= 0) cuda_free(net.workspace);
else free(net.workspace);
if (net.input_state_gpu) cuda_free(net.input_state_gpu);
if (*net.input_gpu) cuda_free(*net.input_gpu);
if (*net.truth_gpu) cuda_free(*net.truth_gpu);
if (net.input_gpu) free(net.input_gpu);
@ -866,7 +867,7 @@ void calculate_binary_weights(network net)
binary_align_weights(l);
if(net.layers[j + 1].use_bin_output) {
if(net.layers[j].use_bin_output) {
l->activation = LINEAR;
}
}