wip 8x4 sparseness

This commit is contained in:
Jean-Marc Valin 2020-12-19 19:25:59 -05:00
parent 8e405b44e0
commit cc28518699
3 changed files with 13 additions and 9 deletions

View file

@ -102,7 +102,7 @@ del pred
del in_exc
# dump models to disk as we go
checkpoint = ModelCheckpoint('lpcnet32c_384_10_G16_{epoch:02d}.h5')
checkpoint = ModelCheckpoint('lpcnet32v_384_10_G16_{epoch:02d}.h5')
#Set this to True to adapt an existing model (e.g. on new data)
adaptation = False
@ -120,5 +120,5 @@ else:
decay = 5e-5
model.compile(optimizer=Adam(lr, decay=decay, beta_2=0.99), loss='sparse_categorical_crossentropy')
model.save_weights('lpcnet32c_384_10_G16_00.h5');
model.save_weights('lpcnet32v_384_10_G16_00.h5');
model.fit([in_data, features, periods], out_exc, batch_size=batch_size, epochs=nb_epochs, validation_split=0.0, callbacks=[checkpoint, sparsify])