Joseph Redmon
2016-09-25 89b0c687b2dd9fff1a7946b24bfeeb0d12837308
src/network.h
@@ -65,6 +65,8 @@
} network_state;
#ifdef GPU
float train_networks(network *nets, int n, data d, int interval);
void sync_nets(network *nets, int n, int interval);
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);