-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathreadme.txt
30 lines (23 loc) · 945 Bytes
/
readme.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
"""
Created on 30/8/2018
@author: kanwang
"""
Function:
dataset.py: read the data into the model
Tasnet_model.py:the forword network
Tasnet_train.py:the main function to run
trainer.py:calculate the loss and for training and testing
utils.py: process the raw audio and other useful functions
train.yaml: all the parameters used in the model
test.py:separate the mixed audio and calculate SDR
Training stage:
1.from the beginning: remove the line with "trainer.rerun" in Tasnet_train.py, use "trainer.run" instead
2.from a trained model: remove the line with "trainer.run" in Tasnet_train.py,use "trainer.rerun" instead, and change the "model_path" in train.yaml/temp
command:python Tasnet_train.py
Testing:
change the parameters under train.yaml/test
command:python test.py
model:
the trained model
loss:
record the process of training