From f1ad9a0487dc60c01e56bacb2a67de1e1006b4b7 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 05 Apr 2018 22:31:04 +0000
Subject: [PATCH] Compile fix
---
src/network.c | 10 +++++++---
1 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/src/network.c b/src/network.c
index bfade3c..438829a 100644
--- a/src/network.c
+++ b/src/network.c
@@ -757,7 +757,7 @@
layer *l = &net.layers[j];
if (l->type == CONVOLUTIONAL) {
- printf(" Fuse Convolutional layer \t\t l->size = %d \n", l->size);
+ //printf(" Merges Convolutional-%d and batch_norm \n", j);
if (l->batch_normalize) {
int f;
@@ -775,11 +775,15 @@
}
l->batch_normalize = 0;
- push_convolutional_layer(*l);
+#ifdef GPU
+ if (gpu_index >= 0) {
+ push_convolutional_layer(*l);
+ }
+#endif
}
}
else {
- printf(" Skip layer: %d \n", l->type);
+ //printf(" Fusion skip layer type: %d \n", l->type);
}
}
}
--
Gitblit v1.10.0