mirror of
https://github.com/xiph/opus.git
synced 2025-05-28 06:09:15 +00:00
Adding option to change frame rate network size
This commit is contained in:
parent
fe7b54c0e8
commit
b24e53fdfa
3 changed files with 10 additions and 8 deletions
|
@ -249,8 +249,9 @@ filename = sys.argv[1]
|
|||
with h5py.File(filename, "r") as f:
|
||||
units = min(f['model_weights']['gru_a']['gru_a']['recurrent_kernel:0'].shape)
|
||||
units2 = min(f['model_weights']['gru_b']['gru_b']['recurrent_kernel:0'].shape)
|
||||
cond_size = min(f['model_weights']['feature_dense1']['feature_dense1']['kernel:0'].shape)
|
||||
|
||||
model, _, _ = lpcnet.new_lpcnet_model(rnn_units1=units, rnn_units2=units2, flag_e2e = flag_e2e)
|
||||
model, _, _ = lpcnet.new_lpcnet_model(rnn_units1=units, rnn_units2=units2, flag_e2e = flag_e2e, cond_size=cond_size)
|
||||
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy'])
|
||||
#model.summary()
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue