From 2ce6460c79e06caa33eab3991ee3e7fd9f0909d6 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Tue, 10 Apr 2018 18:08:24 +0000
Subject: [PATCH] Remove truth only if smaller than 1x1 pix during training
---
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