Fixed bug for 32-bit compilation without GPU.

This commit is contained in:
AlexeyAB
2018-10-23 21:59:57 +03:00
parent d487bdf471
commit 31ac46ba22
5 changed files with 56 additions and 24 deletions

View File

@ -615,7 +615,7 @@ void binary_align_weights(convolutional_layer *l)
binarize_weights(l->weights, m, k, l->binary_weights);
size_t align_weights_size = new_lda * m;
l->align_bit_weights_size = align_weights_size / 8;// +1;
l->align_bit_weights_size = align_weights_size / 8 + 1;
float *align_weights = calloc(align_weights_size, sizeof(float));
l->align_bit_weights = calloc(l->align_bit_weights_size, sizeof(char));