| | |
| | | int out_h, out_w, out_c; |
| | | int n; |
| | | int max_boxes; |
| | | int small_object; |
| | | int groups; |
| | | int size; |
| | | int side; |
| | |
| | | int tanh; |
| | | int *mask; |
| | | int total; |
| | | float bflops; |
| | | |
| | | int adam; |
| | | float B1; |
| | |
| | | float coord_scale; |
| | | float object_scale; |
| | | float noobject_scale; |
| | | float mask_scale; |
| | | float class_scale; |
| | | int bias_match; |
| | | int random; |
| | |
| | | float *weights; |
| | | float *weight_updates; |
| | | |
| | | char *align_bit_weights; |
| | | float *mean_arr; |
| | | int lda_align; |
| | | |
| | | float *col_image; |
| | | int * input_layers; |
| | | int * input_sizes; |
| | |
| | | #ifdef CUDNN |
| | | cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc; |
| | | cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc; |
| | | cudnnTensorDescriptor_t normTensorDesc; |
| | | cudnnTensorDescriptor_t normTensorDesc, normDstTensorDesc, normDstTensorDescF16; |
| | | cudnnFilterDescriptor_t weightDesc; |
| | | cudnnFilterDescriptor_t dweightDesc; |
| | | cudnnConvolutionDescriptor_t convDesc; |