| | |
| | | //printf("%ld\n", workspace_size); |
| | | #ifdef GPU |
| | | if(gpu_index >= 0){ |
| | | net.workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1); |
| | | net.workspace = cuda_make_array(0, workspace_size/sizeof(float) + 1); |
| | | }else { |
| | | net.workspace = calloc(1, workspace_size); |
| | | } |
| | |
| | | net.workspace = calloc(1, workspace_size); |
| | | #endif |
| | | } |
| | | LAYER_TYPE lt = net.layers[net.n - 1].type; |
| | | if ((net.w % 32 != 0 || net.h % 32 != 0) && (lt == YOLO || lt == REGION || lt == DETECTION)) { |
| | | printf("\n Warning: width=%d and height=%d in cfg-file must be divisible by 32 for default networks Yolo v1/v2/v3!!! \n\n", |
| | | net.w, net.h); |
| | | } |
| | | return net; |
| | | } |
| | | |