From 881d6ee9b6625ee502cb4f27d9b017a3da78caa7 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 13 May 2016 20:46:31 +0000
Subject: [PATCH] fixed
---
src/convolutional_layer.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index a93087f..303f1ef 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -155,6 +155,7 @@
l.inputs = l.w * l.h * l.c;
l.col_image = calloc(out_h*out_w*size*size*c, sizeof(float));
+ l.workspace_size = out_h*out_w*size*size*c*sizeof(float);
l.output = calloc(l.batch*out_h * out_w * n, sizeof(float));
l.delta = calloc(l.batch*out_h * out_w * n, sizeof(float));
@@ -188,7 +189,6 @@
l.scales_gpu = cuda_make_array(l.scales, n);
l.scale_updates_gpu = cuda_make_array(l.scale_updates, n);
- l.workspace_size = out_h*out_w*size*size*c;
l.delta_gpu = cuda_make_array(l.delta, l.batch*out_h*out_w*n);
l.output_gpu = cuda_make_array(l.output, l.batch*out_h*out_w*n);
--
Gitblit v1.10.0