ADAM

Joseph Redmon

2016-10-26 352ae7e65b6a74bcd768aa88b866a44c713284c8

 src/layer.h

@@ -94,6 +94,14 @@
    int reorg;
    int log;

    int adam;
    float B1;
    float B2;
    float eps;
    float *m_gpu;
    float *v_gpu;
    int t;

    tree *softmax_tree;

    float alpha;