Merge pull request #766 from HotChick91/AlexeyAB-mask

Alexey

2018-08-15 a723e1c62a27aeb39aaf7fcdeb3beb4e89fba32d

 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

}