src/network.h
@@ -34,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; @@ -57,6 +65,7 @@ } network_state; #ifdef GPU float train_networks(network *nets, int n, data d); float train_network_datum_gpu(network net, float *x, float *y); float *network_predict_gpu(network net, float *input); float * get_network_output_gpu_layer(network net, int i);