mirror of
https://github.com/xiph/opus.git
synced 2025-05-19 18:08:29 +00:00
Training seems to work
This commit is contained in:
parent
785a2b2e84
commit
87cd75f6f4
2 changed files with 14 additions and 4 deletions
|
@ -41,7 +41,7 @@ class PCMInit(Initializer):
|
|||
}
|
||||
|
||||
def new_wavernn_model():
|
||||
pcm = Input(shape=(None, 1))
|
||||
pcm = Input(shape=(None, 2))
|
||||
pitch = Input(shape=(None, 1))
|
||||
feat = Input(shape=(None, nb_used_features))
|
||||
dec_feat = Input(shape=(None, 32))
|
||||
|
@ -61,7 +61,7 @@ def new_wavernn_model():
|
|||
cpitch = pitch
|
||||
|
||||
embed = Embedding(256, 128, embeddings_initializer=PCMInit())
|
||||
cpcm = Reshape((-1, 128))(embed(pcm))
|
||||
cpcm = Reshape((-1, 128*2))(embed(pcm))
|
||||
|
||||
|
||||
cfeat = fconv2(fconv1(feat))
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue