| | |
| | | |
| | | #include "activations.h" |
| | | |
| | | struct layer; |
| | | typedef struct layer layer; |
| | | |
| | | typedef enum { |
| | | CONVOLUTIONAL, |
| | | DECONVOLUTIONAL, |
| | |
| | | COST, |
| | | NORMALIZATION, |
| | | AVGPOOL, |
| | | LOCAL |
| | | LOCAL, |
| | | SHORTCUT |
| | | } LAYER_TYPE; |
| | | |
| | | typedef enum{ |
| | | SSE, MASKED |
| | | } COST_TYPE; |
| | | |
| | | typedef struct { |
| | | struct layer{ |
| | | LAYER_TYPE type; |
| | | ACTIVATION activation; |
| | | COST_TYPE cost_type; |
| | | int batch_normalize; |
| | | int batch; |
| | | int forced; |
| | | int flipped; |
| | | int inputs; |
| | | int outputs; |
| | | int truths; |
| | |
| | | int crop_height; |
| | | int sqrt; |
| | | int flip; |
| | | int index; |
| | | float angle; |
| | | float jitter; |
| | | float saturation; |
| | |
| | | float * squared_gpu; |
| | | float * norms_gpu; |
| | | #endif |
| | | } layer; |
| | | }; |
| | | |
| | | void free_layer(layer); |
| | | |