mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
normalization layer
This commit is contained in:
11
src/layer.h
11
src/layer.h
@ -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;
|
||||
|
||||
|
Reference in New Issue
Block a user