Changeset c6e51cd


Ignore:
Timestamp:
Jan 10, 2019, 7:14:31 PM (5 years ago)
Author:
Paul Brossier <piem@piem.org>
Branches:
feature/crepe_org
Children:
378852b
Parents:
b0b167a
Message:

[pitch_crepe] add activations

File:
1 edited

Legend:

Unmodified
Added
Removed
  • src/pitch/pitch_crepe.c

    rb0b167a rc6e51cd  
    3737#include "fmat.h"
    3838#include "ai/tensor.h"
     39#include "ai/activation.h"
    3940#include "ai/conv1d.h"
    4041#include "ai/maxpool1d.h"
     
    319320    aubio_conv1d_do(o->conv_layers[i], block_input,
    320321        o->conv_output[i]);
     322    // relu activation
     323    aubio_activation_relu(o->conv_output[i]);
    321324    aubio_batchnorm_do(o->batchnorm_layers[i], o->conv_output[i],
    322325        o->batchnorm_output[i]);
     
    335338  // compute dense layer
    336339  aubio_dense_do(o->dense_layer, o->flattened, o->dense_output);
     340
     341  // sigmoid activation
     342  aubio_activation_sigmoid(o->dense_output);
    337343
    338344#if 0
Note: See TracChangeset for help on using the changeset viewer.