fix memory free bug

pull/4302/head
dccho 6 years ago
parent dd34fe156a
commit b4c0fbaec8
  1. 6
      src/convolutional_layer.c

@ -769,8 +769,10 @@ void resize_convolutional_layer(convolutional_layer *l, int w, int h)
l->binary_input_gpu = cuda_make_array(0, l->inputs*l->batch);
}
cuda_free(l->activation_input_gpu);
if (l->activation == SWISH || l->activation == MISH) l->activation_input_gpu = cuda_make_array(l->activation_input, total_batch*l->outputs);
if (l->activation == SWISH || l->activation == MISH) {
cuda_free(l->activation_input_gpu);
l->activation_input_gpu = cuda_make_array(l->activation_input, total_batch*l->outputs);
}
}
#ifdef CUDNN
cudnn_convolutional_setup(l, cudnn_fastest);

Loading…
Cancel
Save