normalization layer

This commit is contained in:
Joseph Redmon
2015-07-09 15:22:14 -07:00
parent 42d7438879
commit 75db98db25
13 changed files with 294 additions and 17 deletions

View File

@ -13,7 +13,8 @@ typedef enum {
DROPOUT,
CROP,
ROUTE,
COST
COST,
NORMALIZATION
} LAYER_TYPE;
typedef enum{
@ -48,6 +49,10 @@ typedef struct {
int does_cost;
int joint;
float alpha;
float beta;
float kappa;
int dontload;
float probability;
@ -69,6 +74,8 @@ typedef struct {
int * input_sizes;
float * delta;
float * output;
float * squared;
float * norms;
#ifdef GPU
int *indexes_gpu;
@ -86,6 +93,8 @@ typedef struct {
float * output_gpu;
float * delta_gpu;
float * rand_gpu;
float * squared_gpu;
float * norms_gpu;
#endif
} layer;