src/network.h
@@ -2,8 +2,13 @@ #ifndef NETWORK_H #define NETWORK_H #include "image.h" #include "layer.h" #ifdef __cplusplus extern "C" { #endif #include "image.h" #include "data.h" #include "tree.h" @@ -37,6 +42,11 @@ int num_steps; int burn_in; int adam; float B1; float B2; float eps; int inputs; int h, w, c; int max_crop; @@ -120,5 +130,9 @@ int get_network_nuisance(network net); int get_network_background(network net); #ifdef __cplusplus } #endif #endif