From: Paul Brossier Date: Thu, 17 Jan 2019 21:39:09 +0000 (+0100) Subject: [batchnorm] remove train for now X-Git-Url: https://git.aubio.org/?a=commitdiff_plain;h=49688ef6b474073b6abc337b36c98067cea6fe77;p=aubio.git [batchnorm] remove train for now --- diff --git a/src/ai/batchnorm.h b/src/ai/batchnorm.h index 7ef55c89..fb89174f 100644 --- a/src/ai/batchnorm.h +++ b/src/ai/batchnorm.h @@ -46,8 +46,6 @@ void aubio_batchnorm_do(aubio_batchnorm_t *t, aubio_tensor_t *input_tensor, aubio_tensor_t *activations); -void aubio_batchnorm_train(aubio_batchnorm_t *t, aubio_tensor_t *input); - uint_t aubio_batchnorm_set_gamma(aubio_batchnorm_t *t, fvec_t *gamma); uint_t aubio_batchnorm_set_beta(aubio_batchnorm_t *t, fvec_t *beta); uint_t aubio_batchnorm_set_moving_mean(aubio_batchnorm_t *t, fvec_t *moving_mean);