Changeset c6e51cd

Timestamp:

Jan 10, 2019, 7:14:31 PM (6 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe_org

Children:

378852b

Parents:

b0b167a

Message:

[pitch_crepe] add activations

File:

• src/pitch/pitch_crepe.c (modified) (3 diffs)

src/pitch/pitch_crepe.c

@@ -37 +37 @@
 #include "fmat.h"
 #include "ai/tensor.h"
+#include "ai/activation.h"
 #include "ai/conv1d.h"
 #include "ai/maxpool1d.h"
@@ -319 +320 @@
     aubio_conv1d_do(o->conv_layers[i], block_input,
         o->conv_output[i]);
+    // relu activation
+    aubio_activation_relu(o->conv_output[i]);
     aubio_batchnorm_do(o->batchnorm_layers[i], o->conv_output[i],
         o->batchnorm_output[i]);
@@ -335 +338 @@
   // compute dense layer
   aubio_dense_do(o->dense_layer, o->flattened, o->dense_output);
+
+  // sigmoid activation
+  aubio_activation_sigmoid(o->dense_output);
 
 #if 0