eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-05-21 19:08:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Making dump_lpcnet.py a bit more robust

Browse source 
Avoid relying on the order of the layers
This commit is contained in:
Jean-Marc Valin 2019-01-17 17:03:48 -05:00
parent d181139930
commit 4698b28345
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
dnn/dump_lpcnet.py
View file

@ -236,16 +236,16 @@ hf.write('#ifndef RNN_DATA_H\n#define RNN_DATA_H\n\n#include "nnet.h"\n\n')
embed_size = lpcnet.embed_size
E = model.get_layer('embed_sig').get_weights()[0]
W = model.layers[18].get_weights()[0][:embed_size,:]
W = model.get_layer('gru_a').get_weights()[0][:embed_size,:]
dump_embedding_layer_impl('gru_a_embed_sig', np.dot(E, W), f, hf)
W = model.layers[18].get_weights()[0][embed_size:2*embed_size,:]
W = model.get_layer('gru_a').get_weights()[0][embed_size:2*embed_size,:]
dump_embedding_layer_impl('gru_a_embed_pred', np.dot(E, W), f, hf)
E = model.get_layer('embed_exc').get_weights()[0]
W = model.layers[18].get_weights()[0][2*embed_size:3*embed_size,:]
W = model.get_layer('gru_a').get_weights()[0][2*embed_size:3*embed_size,:]
dump_embedding_layer_impl('gru_a_embed_exc', np.dot(E, W), f, hf)
W = model.layers[18].get_weights()[0][3*embed_size:,:]
W = model.get_layer('gru_a').get_weights()[0][3*embed_size:,:]
#FIXME: dump only half the biases
b = model.layers[18].get_weights()[2]
b = model.get_layer('gru_a').get_weights()[2]
dump_dense_layer_impl('gru_a_dense_feature', W, b, 'LINEAR', f, hf)
layer_list = []