10 class weight

Edmond Yoo

2018-09-16 f3aced12fcc41415851ffb463675930cea8dd086

 src/layer.h

@@ -113,6 +113,7 @@
   int tanh;
   int *mask;
   int total;
   float bflops;

    int adam;
    float B1;
@@ -178,6 +179,11 @@
    float *weights;
    float *weight_updates;

    char *align_bit_weights;
    float *mean_arr;
    int lda_align;
    int bit_align;

    float *col_image;
    int   * input_layers;
    int   * input_sizes;
@@ -281,7 +287,7 @@
    #ifdef CUDNN
    cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
    cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc, normDstTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc, normDstTensorDesc, normDstTensorDescF16;
    cudnnFilterDescriptor_t weightDesc;
    cudnnFilterDescriptor_t dweightDesc;
    cudnnConvolutionDescriptor_t convDesc;