From a723e1c62a27aeb39aaf7fcdeb3beb4e89fba32d Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Wed, 15 Aug 2018 20:52:09 +0000
Subject: [PATCH] Merge pull request #766 from HotChick91/AlexeyAB-mask
---
src/shortcut_layer.c | 24 ++++++++++++------------
1 files changed, 12 insertions(+), 12 deletions(-)
diff --git a/src/shortcut_layer.c b/src/shortcut_layer.c
index 01e133b..9fa18db 100644
--- a/src/shortcut_layer.c
+++ b/src/shortcut_layer.c
@@ -38,20 +38,20 @@
void resize_shortcut_layer(layer *l, int w, int h)
{
- assert(l->w == l->out_w);
- assert(l->h == l->out_h);
- l->w = l->out_w = w;
- l->h = l->out_h = h;
- l->outputs = w*h*l->out_c;
- l->inputs = l->outputs;
- l->delta = realloc(l->delta, l->outputs*l->batch * sizeof(float));
- l->output = realloc(l->output, l->outputs*l->batch * sizeof(float));
+ //assert(l->w == l->out_w);
+ //assert(l->h == l->out_h);
+ l->w = l->out_w = w;
+ l->h = l->out_h = h;
+ l->outputs = w*h*l->out_c;
+ l->inputs = l->outputs;
+ l->delta = realloc(l->delta, l->outputs*l->batch * sizeof(float));
+ l->output = realloc(l->output, l->outputs*l->batch * sizeof(float));
#ifdef GPU
- cuda_free(l->output_gpu);
- cuda_free(l->delta_gpu);
- l->output_gpu = cuda_make_array(l->output, l->outputs*l->batch);
- l->delta_gpu = cuda_make_array(l->delta, l->outputs*l->batch);
+ cuda_free(l->output_gpu);
+ cuda_free(l->delta_gpu);
+ l->output_gpu = cuda_make_array(l->output, l->outputs*l->batch);
+ l->delta_gpu = cuda_make_array(l->delta, l->outputs*l->batch);
#endif
}
--
Gitblit v1.10.0