From 96c629d069e8811710a1eb587c1d63c65dd3131f Mon Sep 17 00:00:00 2001 From: Soonhwan-Kwon Date: Fri, 7 Jul 2017 01:01:15 +0900 Subject: [PATCH] roll back original architecture of arch_deepspeech.py --- arch_deepspeech.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/arch_deepspeech.py b/arch_deepspeech.py index 82ffe12..4288b24 100644 --- a/arch_deepspeech.py +++ b/arch_deepspeech.py @@ -115,9 +115,9 @@ def arch(args, seq_len=None): stride=conv_layer2_stride, no_bias=is_batchnorm, name='conv2') - # if is_batchnorm: - # # batch norm normalizes axis 1 - # net = batchnorm(net, name="conv2_batchnorm") + if is_batchnorm: + # batch norm normalizes axis 1 + net = batchnorm(net, name="conv2_batchnorm") net = mx.sym.transpose(data=net, axes=(0, 2, 1, 3)) net = mx.sym.Reshape(data=net, shape=(0, 0, -3))