Skip to content

Commit

Permalink
Compile fix for CPP
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexeyAB committed Aug 5, 2020
1 parent 10fdf9a commit cfcb37a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/softmax_layer.c
Original file line number Diff line number Diff line change
Expand Up @@ -197,7 +197,7 @@ contrastive_layer make_contrastive_layer(int batch, int w, int h, int c, int cla

const int max_contr_size = (l.max_boxes*l.batch)*(l.max_boxes*l.batch) * sizeof(contrastive_params)/4;
printf(" max_contr_size = %d MB \n", max_contr_size / (1024*1024));
l.contrast_p_gpu = cuda_make_array(NULL, max_contr_size);
l.contrast_p_gpu = (contrastive_params *)cuda_make_array(NULL, max_contr_size);
#endif
fprintf(stderr, "contrastive %4d x%4d x%4d x emb_size %4d x batch: %4d classes = %4d, step = %4d \n", w, h, l.n, l.embedding_size, batch, l.classes, step);
if(l.detection) fprintf(stderr, "detection \n");
Expand Down

0 comments on commit cfcb37a

Please sign in to comment.