From 6b8fd6f33f6a61138136fd022c2b887ae39e2c42 Mon Sep 17 00:00:00 2001 From: Alexey <AlexeyAB@users.noreply.github.com> Date: Sun, 06 May 2018 23:18:06 +0000 Subject: [PATCH] Update Readme.md --- 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