mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Added Swish-activation

Browse Source
This commit is contained in:
AlexeyAB
2019-06-26 15:48:22 +03:00
parent 5ec35922d5
commit 88cccfcad4
7 changed files with 93 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/darknet.h
View File

@ -102,7 +102,7 @@ typedef struct tree {
// activations.h
typedef enum {
LOGISTIC, RELU, RELIE, LINEAR, RAMP, TANH, PLSE, LEAKY, ELU, LOGGY, STAIR, HARDTAN, LHTAN, SELU
LOGISTIC, RELU, RELIE, LINEAR, RAMP, TANH, PLSE, LEAKY, ELU, LOGGY, STAIR, HARDTAN, LHTAN, SELU, SWISH
}ACTIVATION;
// parser.h
@ -339,6 +339,7 @@ struct layer {
float *col_image;
float * delta;
float * output;
float * output_sigmoid;
int delta_pinned;
int output_pinned;
float * loss;
@ -522,6 +523,7 @@ struct layer {
float * scale_change_gpu;
float * output_gpu;
float * output_sigmoid_gpu;
float * loss_gpu;
float * delta_gpu;
float * rand_gpu;