Compile fix

pull/2383/head
AlexeyAB 6 years ago
parent 5448e07445
commit 00e992a600
  1. 2
      src/network.c

@ -997,6 +997,7 @@ void calculate_binary_weights(network net)
l->activation = LINEAR; l->activation = LINEAR;
} }
#ifdef GPU
// fuse conv_xnor + shortcut -> conv_xnor // fuse conv_xnor + shortcut -> conv_xnor
if ((j + 1) < net.n && net.layers[j].type == CONVOLUTIONAL) { if ((j + 1) < net.n && net.layers[j].type == CONVOLUTIONAL) {
layer *sc = &net.layers[j + 1]; layer *sc = &net.layers[j + 1];
@ -1009,6 +1010,7 @@ void calculate_binary_weights(network net)
net.layers[j + 1].forward_gpu = forward_blank_layer; net.layers[j + 1].forward_gpu = forward_blank_layer;
} }
} }
#endif // GPU
} }
} }
} }

Loading…
Cancel
Save