From 6d56c38e8bcb9041335b03f27c192c24dfaedb1c Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Wed, 28 Mar 2018 23:39:28 +0000 Subject: [PATCH] Merge branch 'master' of github.com:AlexeyAB/darknet --- src/layer.c | 2 ++ 1 files changed, 2 insertions(+), 0 deletions(-) diff --git a/src/layer.c b/src/layer.c index b88c941..582cbb3 100644 --- a/src/layer.c +++ b/src/layer.c @@ -83,6 +83,8 @@ if (l.x_norm_gpu) cuda_free(l.x_norm_gpu); if (l.weights_gpu) cuda_free(l.weights_gpu); if (l.weight_updates_gpu) cuda_free(l.weight_updates_gpu); + if (l.weights_gpu16) cuda_free(l.weights_gpu16); + if (l.weight_updates_gpu16) cuda_free(l.weight_updates_gpu16); if (l.biases_gpu) cuda_free(l.biases_gpu); if (l.bias_updates_gpu) cuda_free(l.bias_updates_gpu); if (l.scales_gpu) cuda_free(l.scales_gpu); -- Gitblit v1.10.0