From 3459c47bfa02d23d300df9a44ba3ec9c958850fe Mon Sep 17 00:00:00 2001 From: Philip Kahn <tigerhawkvok@gmail.com> Date: Fri, 04 May 2018 21:00:11 +0000 Subject: [PATCH] Also replace root version to Python3 --- 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