Alexey
2017-03-01 a360f694093f7748e3232f4ed74d40446d735fc3
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,11 @@
    float object_scale;
    float noobject_scale;
    float class_scale;
    int bias_match;
    int random;
    float thresh;
    int classfix;
    int absolute;
    int dontload;
    int dontloadscales;