From be9d971ddb9ea0520da78cfff7788eb5481f095e Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Tue, 03 Apr 2018 13:56:53 +0000
Subject: [PATCH] Compile error fix
---
src/network.c | 6 +++++-
1 files changed, 5 insertions(+), 1 deletions(-)
diff --git a/src/network.c b/src/network.c
index bfade3c..d532b85 100644
--- a/src/network.c
+++ b/src/network.c
@@ -775,7 +775,11 @@
}
l->batch_normalize = 0;
- push_convolutional_layer(*l);
+#ifdef GPU
+ if (gpu_index >= 0) {
+ push_convolutional_layer(*l);
+ }
+#endif
}
}
else {
--
Gitblit v1.10.0