src/layer.h
@@ -3,6 +3,7 @@ #include "activations.h" #include "stddef.h" #include "tree.h" struct network_state; @@ -66,6 +67,7 @@ int size; int side; int stride; int reverse; int pad; int sqrt; int flip; @@ -93,6 +95,19 @@ int reorg; int log; int adam; float B1; float B2; float eps; float *m_gpu; float *v_gpu; int t; float *m; float *v; tree *softmax_tree; int *map; float alpha; float beta; float kappa; @@ -101,7 +116,10 @@ float object_scale; float noobject_scale; float class_scale; int bias_match; int random; float thresh; int classfix; int dontload; int dontloadscales;
