Faster batch normalization

Joseph Redmon

2015-12-08 c2738835f0a2435ab03f411af3d168aec389d2a6

 src/layer.h

@@ -15,7 +15,8 @@
    ROUTE,
    COST,
    NORMALIZATION,
    AVGPOOL
    AVGPOOL,
    LOCAL
} LAYER_TYPE;

typedef enum{
@@ -129,11 +130,12 @@
    float * x_gpu;
    float * x_norm_gpu;
    float * weights_gpu;
    float * biases_gpu;
    float * scales_gpu;

    float * weight_updates_gpu;

    float * biases_gpu;
    float * bias_updates_gpu;

    float * scales_gpu;
    float * scale_updates_gpu;

    float * output_gpu;