Skip to content

Commit fc78680

Browse files
committed
Adding instructions on training.
1 parent 043f2d9 commit fc78680

File tree

2 files changed

+13
-2
lines changed

2 files changed

+13
-2
lines changed

TRAINING

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
(1) cd src ; ./compile.sh
2+
3+
(2) ./denoise_training signal.raw noise.raw count > training.f32
4+
5+
(note the matrix size and replace 500000 87 below)
6+
7+
(3) cd training ; ./bin2hdf5.py ../src/training.f32 500000 87 training.h5
8+
9+
(4) ./rnn_train.py
10+
11+
(5) ./dump_rnn.py weights.hdf5 ../src/rnn_data.c ../src/rnn_data.h

training/rnn_train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def get_config(self):
8282
batch_size = 32
8383

8484
print('Loading data...')
85-
with h5py.File('denoise_data9.h5', 'r') as hf:
85+
with h5py.File('training.h5', 'r') as hf:
8686
all_data = hf['data'][:]
8787
print('done.')
8888

@@ -113,4 +113,4 @@ def get_config(self):
113113
batch_size=batch_size,
114114
epochs=120,
115115
validation_split=0.1)
116-
model.save("newweights9i.hdf5")
116+
model.save("weights.hdf5")

0 commit comments

Comments
 (0)