From a99050f0c8cb0315fa31e3d1fa3e38594fe5e40a Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 08 Dec 2014 04:16:21 +0000
Subject: [PATCH] Some fixes to momentum
---
src/network_gpu.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/network_gpu.c b/src/network_gpu.c
index 938a014..fe53d0c 100644
--- a/src/network_gpu.c
+++ b/src/network_gpu.c
@@ -87,7 +87,7 @@
}
if(net.types[i] == CONVOLUTIONAL){
convolutional_layer layer = *(convolutional_layer *)net.layers[i];
- backward_convolutional_layer_gpu(layer, prev_delta);
+ backward_convolutional_layer_gpu(layer, prev_input, prev_delta);
}
else if(net.types[i] == COST){
cost_layer layer = *(cost_layer *)net.layers[i];
--
Gitblit v1.10.0