mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
probably how maxpool layers should be
This commit is contained in:
15
src/dropout_layer.h
Normal file
15
src/dropout_layer.h
Normal file
@ -0,0 +1,15 @@
|
||||
#ifndef DROPOUT_LAYER_H
|
||||
#define DROPOUT_LAYER_H
|
||||
|
||||
typedef struct{
|
||||
int batch;
|
||||
int inputs;
|
||||
float probability;
|
||||
} dropout_layer;
|
||||
|
||||
dropout_layer *make_dropout_layer(int batch, int inputs, float probability);
|
||||
|
||||
void forward_dropout_layer(dropout_layer layer, float *input);
|
||||
void backward_dropout_layer(dropout_layer layer, float *input, float *delta);
|
||||
|
||||
#endif
|
Reference in New Issue
Block a user