From 5c067dc44785a761a0243d8cd634e3ac17d548ad Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 12 Sep 2016 20:55:20 +0000
Subject: [PATCH] good chance I didn't break anything
---
src/layer.c | 6 ++----
1 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/src/layer.c b/src/layer.c
index 557aa3b..ccd0daf 100644
--- a/src/layer.c
+++ b/src/layer.c
@@ -14,8 +14,6 @@
if(l.indexes) free(l.indexes);
if(l.rand) free(l.rand);
if(l.cost) free(l.cost);
- if(l.filters) free(l.filters);
- if(l.filter_updates) free(l.filter_updates);
if(l.biases) free(l.biases);
if(l.bias_updates) free(l.bias_updates);
if(l.weights) free(l.weights);
@@ -30,8 +28,8 @@
#ifdef GPU
if(l.indexes_gpu) cuda_free((float *)l.indexes_gpu);
- if(l.filters_gpu) cuda_free(l.filters_gpu);
- if(l.filter_updates_gpu) cuda_free(l.filter_updates_gpu);
+ if(l.weights_gpu) cuda_free(l.weights_gpu);
+ if(l.weight_updates_gpu) cuda_free(l.weight_updates_gpu);
if(l.col_image_gpu) cuda_free(l.col_image_gpu);
if(l.weights_gpu) cuda_free(l.weights_gpu);
if(l.biases_gpu) cuda_free(l.biases_gpu);
--
Gitblit v1.10.0