From 677182fcaad641dae4a988861713c9b8fe2f9282 Mon Sep 17 00:00:00 2001 From: Jean-Marc Valin Date: Sat, 29 Dec 2018 15:41:59 -0500 Subject: [PATCH] Making the update and reset gate more sparse --- dnn/train_lpcnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dnn/train_lpcnet.py b/dnn/train_lpcnet.py index 6f445c7e..837778f0 100755 --- a/dnn/train_lpcnet.py +++ b/dnn/train_lpcnet.py @@ -101,4 +101,4 @@ checkpoint = ModelCheckpoint('lpcnet18_384_10_G16_{epoch:02d}.h5') model.load_weights('lpcnet9b_384_10_G16_01.h5') model.compile(optimizer=Adam(0.001, amsgrad=True, decay=5e-5), loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy']) -model.fit([in_data, in_exc, features, periods], out_exc, batch_size=batch_size, epochs=nb_epochs, validation_split=0.0, callbacks=[checkpoint, lpcnet.Sparsify(2000, 40000, 400, (0.1, 0.1, 0.1))]) +model.fit([in_data, in_exc, features, periods], out_exc, batch_size=batch_size, epochs=nb_epochs, validation_split=0.0, callbacks=[checkpoint, lpcnet.Sparsify(2000, 40000, 400, (0.05, 0.05, 0.2))])