From 9a01e6ccb7a74ff77e99060cf18acd6cfdb74b8e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 11 Nov 2016 16:48:40 +0000
Subject: [PATCH] :fire: crush. crush. admit. :fire:
---
src/convolutional_layer.c | 10 +++++++---
1 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index 888eca3..1d93b3f 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -235,6 +235,11 @@
l.rolling_mean = calloc(n, sizeof(float));
l.rolling_variance = calloc(n, sizeof(float));
}
+ if(adam){
+ l.adam = 1;
+ l.m = calloc(c*n*size*size, sizeof(float));
+ l.v = calloc(c*n*size*size, sizeof(float));
+ }
#ifdef GPU
l.forward_gpu = forward_convolutional_layer_gpu;
@@ -243,9 +248,8 @@
if(gpu_index >= 0){
if (adam) {
- l.adam = 1;
- l.m_gpu = cuda_make_array(l.weight_updates, c*n*size*size);
- l.v_gpu = cuda_make_array(l.weight_updates, c*n*size*size);
+ l.m_gpu = cuda_make_array(l.m, c*n*size*size);
+ l.v_gpu = cuda_make_array(l.v, c*n*size*size);
}
l.weights_gpu = cuda_make_array(l.weights, c*n*size*size);
--
Gitblit v1.10.0