Compile error fix

7 years ago · be9d971ddb
parent 1b2c70f82a
commit be9d971ddb
1 changed files with 5 additions and 1 deletions
--- a/src/network.c
+++ b/src/network.c
@ -775,7 +775,11 @@ void fuse_conv_batchnorm(network net)
 				}

 				l->batch_normalize = 0;
-				push_convolutional_layer(*l);
+#ifdef GPU
+				if (gpu_index >= 0) {
+					push_convolutional_layer(*l);
+				}
+#endif
 			}
 		}
 		else {