stuff

Joseph Redmon

2016-03-14 02bb33c64514ef36d48388e2265b034c49bb31c4

 src/layer.h

@@ -20,11 +20,14 @@
    NORMALIZATION,
    AVGPOOL,
    LOCAL,
    SHORTCUT
    SHORTCUT,
    ACTIVE,
    RNN,
    CRNN
} LAYER_TYPE;

typedef enum{
    SSE, MASKED
    SSE, MASKED, SMOOTH
} COST_TYPE;

struct layer{
@@ -32,6 +35,7 @@
    ACTIVATION activation;
    COST_TYPE cost_type;
    int batch_normalize;
    int shortcut;
    int batch;
    int forced;
    int flipped;
@@ -46,11 +50,13 @@
    int side;
    int stride;
    int pad;
    int crop_width;
    int crop_height;
    int sqrt;
    int flip;
    int index;
    int binary;
    int steps;
    int hidden;
    float dot;
    float angle;
    float jitter;
    float saturation;
@@ -78,6 +84,7 @@
    int dontload;
    int dontloadscales;

    float temperature;
    float probability;
    float scale;

@@ -86,6 +93,9 @@
    float *cost;
    float *filters;
    float *filter_updates;
    float *state;

    float *binary_filters;

    float *biases;
    float *bias_updates;
@@ -108,14 +118,28 @@
    float * mean;
    float * variance;

    float * mean_delta;
    float * variance_delta;

    float * rolling_mean;
    float * rolling_variance;

    float * x;
    float * x_norm;

    struct layer *input_layer;
    struct layer *self_layer;
    struct layer *output_layer;

    #ifdef GPU
    int *indexes_gpu;
    float * state_gpu;
    float * filters_gpu;
    float * filter_updates_gpu;

    float *binary_filters_gpu;
    float *mean_filters_gpu;

    float * spatial_mean_gpu;
    float * spatial_variance_gpu;