src/network.h
@@ -65,7 +65,8 @@ } network_state; #ifdef GPU float train_networks(network *nets, int n, data d); 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);
