src/network.h
@@ -11,6 +11,7 @@ } learning_rate_policy; typedef struct network{ float *workspace; int n; int batch; int *seen; @@ -33,11 +34,19 @@ float *scales; int *steps; int num_steps; int burn_in; int inputs; int h, w, c; int max_crop; int min_crop; float angle; float aspect; float exposure; float saturation; float hue; int gpu_index; #ifdef GPU float **input_gpu; @@ -49,6 +58,7 @@ float *truth; float *input; float *delta; float *workspace; int train; int index; network net;