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

stashing stuff here

Browse source
This commit is contained in:
Jean-Marc Valin 2018-07-12 18:20:25 -04:00
parent 679dfbab58
commit 374ba430c4
4 changed files with 21 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

10
dnn/train_lpcnet.py
View file

@ -13,13 +13,13 @@ from adadiff import Adadiff
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.28
config.gpu_options.per_process_gpu_memory_fraction = 0.44
set_session(tf.Session(config=config))
nb_epochs = 40
batch_size = 64
model = lpcnet.new_wavernn_model()
model, enc, dec = lpcnet.new_wavernn_model()
model.compile(optimizer=Adadiff(), loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy'])
model.summary()
@ -63,8 +63,8 @@ features = features[:, :, :nb_used_features]
# f.create_dataset('data', data=in_data[:50000, :, :])
# f.create_dataset('feat', data=features[:50000, :, :])
checkpoint = ModelCheckpoint('lpcnet1k_{epoch:02d}.h5')
checkpoint = ModelCheckpoint('lpcnet3b_{epoch:02d}.h5')
#model.load_weights('wavernn1c_01.h5')
model.compile(optimizer=Adadiff(), loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy'])
model.fit([in_data, in_pitch, features], out_data, batch_size=batch_size, epochs=30, validation_split=0.2, callbacks=[checkpoint])
model.compile(optimizer=Adam(0.001, amsgrad=True, decay=2e-4), loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy'])
model.fit([in_data, features], out_data, batch_size=batch_size, epochs=30, validation_split=0.2, callbacks=[checkpoint])