Joseph Redmon
2015-07-11 ccde487525fc89a1d4bc3e1cf11a18971e8451c9
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);