Fix conv1d, default to size 384

This commit is contained in:
Jean-Marc Valin 2018-11-24 15:30:17 -05:00
parent 66486004ba
commit c025744e34
2 changed files with 2 additions and 4 deletions

View file

@ -115,7 +115,7 @@ def dump_embedding_layer(self, f, hf):
Embedding.dump_layer = dump_embedding_layer
model, _, _ = lpcnet.new_lpcnet_model(rnn_units1=640, use_gpu=False)
model, _, _ = lpcnet.new_lpcnet_model(rnn_units1=384, use_gpu=False)
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy'])
#model.summary()