Joseph Redmon
2015-07-08 a08ef29e085236f13fd04246c6774f9682a56208
src/network.h
@@ -34,6 +34,8 @@
float * get_network_output_gpu_layer(network net, int i);
float * get_network_delta_gpu_layer(network net, int i);
float *get_network_output_gpu(network net);
void forward_network_gpu(network net, network_state state);
void backward_network_gpu(network net, network_state state);
#endif
void compare_networks(network n1, network n2, data d);
@@ -65,7 +67,7 @@
int get_predicted_class_network(network net);
void print_network(network net);
void visualize_network(network net);
int resize_network(network net, int h, int w, int c);
int resize_network(network *net, int w, int h);
void set_batch_network(network *net, int b);
int get_network_input_size(network net);
float get_network_cost(network net);