src/network.h
@@ -2,8 +2,14 @@ #ifndef NETWORK_H #define NETWORK_H #include "image.h" #include <stdint.h> #include "layer.h" #ifdef __cplusplus extern "C" { #endif #include "image.h" #include "data.h" #include "tree.h" @@ -15,7 +21,7 @@ float *workspace; int n; int batch; int *seen; int *seen; float epoch; int subdivisions; float momentum; @@ -58,6 +64,7 @@ #ifdef GPU float **input_gpu; float **truth_gpu; int wait_stream; #endif } network; @@ -125,5 +132,9 @@ int get_network_nuisance(network net); int get_network_background(network net); #ifdef __cplusplus } #endif #endif
