From abba310982212089d8111f3da661688e9b2f9728 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Wed, 5 Jun 2019 01:04:16 +0300 Subject: [PATCH] Minor fix in cost_layer --- src/cost_layer.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cost_layer.c b/src/cost_layer.c index c967c29d..3038c3da 100644 --- a/src/cost_layer.c +++ b/src/cost_layer.c @@ -51,8 +51,8 @@ cost_layer make_cost_layer(int batch, int inputs, COST_TYPE cost_type, float sca l.forward_gpu = forward_cost_layer_gpu; l.backward_gpu = backward_cost_layer_gpu; - l.delta_gpu = cuda_make_array(l.output, inputs*batch); - l.output_gpu = cuda_make_array(l.delta, inputs*batch); + l.delta_gpu = cuda_make_array(l.delta, inputs*batch); + l.output_gpu = cuda_make_array(l.output, inputs*batch); #endif return l; }