| | |
| | | #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" |
| | | |
| | |
| | | float *workspace; |
| | | int n; |
| | | int batch; |
| | | int *seen; |
| | | uint64_t *seen; |
| | | float epoch; |
| | | int subdivisions; |
| | | float momentum; |
| | |
| | | int num_steps; |
| | | int burn_in; |
| | | |
| | | int adam; |
| | | float B1; |
| | | float B2; |
| | | float eps; |
| | | |
| | | int inputs; |
| | | int h, w, c; |
| | | int max_crop; |
| | |
| | | int get_network_nuisance(network net); |
| | | int get_network_background(network net); |
| | | |
| | | #ifdef __cplusplus |
| | | } |
| | | #endif |
| | | |
| | | #endif |
| | | |