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

so much need to commit

Browse Source
This commit is contained in:
Joseph Redmon
2016-05-06 16:25:16 -07:00
parent 0dff437a69
commit c7b10ceadb
37 changed files with 1502 additions and 438 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/activations.c
View File

@ -28,6 +28,8 @@ char *get_activation_string(ACTIVATION a)
return "plse";
case LEAKY:
return "leaky";
case STAIR:
return "stair";
default:
break;
}
@ -46,6 +48,7 @@ ACTIVATION get_activation(char *s)
if (strcmp(s, "ramp")==0) return RAMP;
if (strcmp(s, "leaky")==0) return LEAKY;
if (strcmp(s, "tanh")==0) return TANH;
if (strcmp(s, "stair")==0) return STAIR;
fprintf(stderr, "Couldn't find activation function %s, going with ReLU\n", s);
return RELU;
}
@ -73,6 +76,8 @@ float activate(float x, ACTIVATION a)
return tanh_activate(x);
case PLSE:
return plse_activate(x);
case STAIR:
return stair_activate(x);
}
return 0;
}
@ -108,6 +113,8 @@ float gradient(float x, ACTIVATION a)
return tanh_gradient(x);
case PLSE:
return plse_gradient(x);
case STAIR:
return stair_gradient(x);
}
return 0;
}