From 60c97b97239d916124f898603cabe7142b41dca0 Mon Sep 17 00:00:00 2001 From: Jean-Marc Valin Date: Sat, 24 Nov 2018 16:19:02 -0500 Subject: [PATCH] Proper ordering for MDense weights --- dnn/dump_lpcnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dnn/dump_lpcnet.py b/dnn/dump_lpcnet.py index 353d250b..f420c92a 100755 --- a/dnn/dump_lpcnet.py +++ b/dnn/dump_lpcnet.py @@ -85,9 +85,9 @@ def dump_mdense_layer(self, f, hf): name = self.name print("printing layer " + name + " of type " + self.__class__.__name__) weights = self.get_weights() - printVector(f, weights[0], name + '_weights') - printVector(f, weights[1], name + '_bias') - printVector(f, weights[1], name + '_factor') + printVector(f, np.transpose(weights[0], (1, 2, 0)), name + '_weights') + printVector(f, np.transpose(weights[1], (1, 0)), name + '_bias') + printVector(f, np.transpose(weights[2], (1, 0)), name + '_factor') activation = self.activation.__name__.upper() max_mdense_tmp = max(max_mdense_tmp, weights[0].shape[0]*weights[0].shape[2]) f.write('const MDenseLayer {} = {{\n {}_bias,\n {}_weights,\n {}_factor,\n {}, {}, {}, ACTIVATION_{}\n}};\n\n'