src/network.h
@@ -4,7 +4,6 @@ #include "image.h" #include "detection_layer.h" #include "params.h" #include "data.h" typedef enum { @@ -17,6 +16,7 @@ NORMALIZATION, DROPOUT, CROP, ROUTE, COST } LAYER_TYPE; @@ -47,6 +47,7 @@ float *network_predict_gpu(network net, float *input); 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); #endif void compare_networks(network n1, network n2, data d);