Joseph Redmon
2014-10-30 2b2441313b73c460a60c013c3b7bf9e19c994b6b
src/network.h
@@ -42,6 +42,7 @@
cl_mem get_network_output_cl_layer(network net, int i);
cl_mem get_network_delta_cl_layer(network net, int i);
float train_network_sgd_gpu(network net, data d, int n);
float train_network_data_gpu(network net, data d, int n);
#endif
network make_network(int n, int batch);