Unverified Commit 9aff6a72 authored by Gregor Richards's avatar Gregor Richards Committed by Jean-Marc Valin
Browse files

Adding instructions on training.

parent 02105923
(1) cd src ; ./compile.sh
(2) ./denoise_training signal.raw noise.raw count > training.f32
(note the matrix size and replace 500000 87 below)
(3) cd training ; ./bin2hdf5.py ../src/training.f32 500000 87 training.h5
(4) ./rnn_train.py
(5) ./dump_rnn.py weights.hdf5 ../src/rnn_data.c ../src/rnn_data.h
......@@ -82,7 +82,7 @@ model.compile(loss=[mycost, my_crossentropy],
batch_size = 32
print('Loading data...')
with h5py.File('denoise_data9.h5', 'r') as hf:
with h5py.File('training.h5', 'r') as hf:
all_data = hf['data'][:]
print('done.')
......@@ -113,4 +113,4 @@ model.fit(x_train, [y_train, vad_train],
batch_size=batch_size,
epochs=120,
validation_split=0.1)
model.save("newweights9i.hdf5")
model.save("weights.hdf5")
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment