From d0b9326a352ed2fbc3ae66fdef40b4533a2f211d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 11 Aug 2015 06:22:27 +0000
Subject: [PATCH] Hacks to get nightmare to not break gridsizing
---
src/detection_layer.c | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/src/detection_layer.c b/src/detection_layer.c
index 6a25819..e48b8b3 100644
--- a/src/detection_layer.c
+++ b/src/detection_layer.c
@@ -204,6 +204,7 @@
backward_detection_layer(l, cpu_state);
cuda_push_array(state.delta, delta_cpu, l.batch*l.inputs);
+ if (truth_cpu) free(truth_cpu);
free(in_cpu);
free(delta_cpu);
}
--
Gitblit v1.10.0