From 28e21152728cbea617948671df064ec75c7953e5 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 07 Dec 2014 08:41:26 +0000
Subject: [PATCH] Distributed training
---
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