stable

Joseph Redmon

2015-09-09 393dc8eb6f3a9dd92ec665200444186c1addc5d2

 src/network.h

@@ -8,7 +8,7 @@
#include "data.h"

typedef enum {
    CONSTANT, STEP, EXP, POLY
    CONSTANT, STEP, EXP, POLY, STEPS, SIG
} learning_rate_policy;

typedef struct {
@@ -25,9 +25,13 @@

    float learning_rate;
    float gamma;
    float scale;
    float power;
    int step;
    int max_batches;
    float *scales;
    int   *steps;
    int num_steps;

    int inputs;
    int h, w, c;