mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Optimized resizing of region_layer for random=1

Browse Source
This commit is contained in:
AlexeyAB
2018-02-21 15:35:09 +03:00
parent 4b0be8c701
commit e4ab47dfce
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/network.c
View File

@ -378,6 +378,7 @@ int resize_network(network *net, int w, int h)
}
#ifdef GPU
if(gpu_index >= 0){
printf(" try to allocate workspace, ");
net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1);
printf(" CUDA allocate done! \n");
}else {

4
src/region_layer.c
View File

@ -53,6 +53,8 @@ region_layer make_region_layer(int batch, int w, int h, int n, int classes, int
void resize_region_layer(layer *l, int w, int h)
{
int old_w = l->w;
int old_h = l->h;
l->w = w;
l->h = h;
@ -63,11 +65,13 @@ void resize_region_layer(layer *l, int w, int h)
l->delta = realloc(l->delta, l->batch*l->outputs*sizeof(float));
#ifdef GPU
if (old_w < w || old_h < h) {
cuda_free(l->delta_gpu);
cuda_free(l->output_gpu);
l->delta_gpu = cuda_make_array(l->delta, l->batch*l->outputs);
l->output_gpu = cuda_make_array(l->output, l->batch*l->outputs);
}
#endif
}