lots of stuff

Joseph Redmon

2016-01-28 913d355ec1cf34aad71fdd75202fc3b0309e63a0

 src/layer.h

@@ -21,11 +21,12 @@
    AVGPOOL,
    LOCAL,
    SHORTCUT,
    ACTIVE
    ACTIVE,
    RNN
} LAYER_TYPE;

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

struct layer{
@@ -50,6 +51,9 @@
    int sqrt;
    int flip;
    int index;
    int binary;
    int steps;
    int hidden;
    float angle;
    float jitter;
    float saturation;
@@ -77,6 +81,7 @@
    int dontload;
    int dontloadscales;

    float temperature;
    float probability;
    float scale;

@@ -85,6 +90,9 @@
    float *cost;
    float *filters;
    float *filter_updates;
    float *state;

    float *binary_filters;

    float *biases;
    float *bias_updates;
@@ -107,14 +115,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;