shortcut layers, msr networks

Joseph Redmon

2015-12-14 db0397cfaaf488364e3d2e1669dfefae2ee6ea73

 src/layer.h

@@ -3,6 +3,9 @@

#include "activations.h"

struct layer;
typedef struct layer layer;

typedef enum {
    CONVOLUTIONAL,
    DECONVOLUTIONAL,
@@ -15,20 +18,23 @@
    ROUTE,
    COST,
    NORMALIZATION,
    AVGPOOL
    AVGPOOL,
    LOCAL,
    SHORTCUT
} LAYER_TYPE;

typedef enum{
    SSE, MASKED
} COST_TYPE;

typedef struct {
struct layer{
    LAYER_TYPE type;
    ACTIVATION activation;
    COST_TYPE cost_type;
    int batch_normalize;
    int batch;
    int forced;
    int flipped;
    int inputs;
    int outputs;
    int truths;
@@ -44,6 +50,7 @@
    int crop_height;
    int sqrt;
    int flip;
    int index;
    float angle;
    float jitter;
    float saturation;
@@ -129,11 +136,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;
@@ -142,7 +150,7 @@
    float * squared_gpu;
    float * norms_gpu;
    #endif
} layer;
};

void free_layer(layer);