src/layer.h
@@ -3,6 +3,7 @@ #include "activations.h" #include "stddef.h" #include "tree.h" struct network_state; @@ -93,6 +94,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 +115,9 @@ float object_scale; float noobject_scale; float class_scale; int bias_match; int random; float thresh; int dontload; int dontloadscales;