speechbrainteam commited on
Commit
db62c92
·
1 Parent(s): b3da5ae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -1
README.md CHANGED
@@ -61,6 +61,31 @@ torchaudio.save("source3hat.wav", est_sources[:, :, 2].detach().cpu(), 8000)
61
  ### Inference on GPU
62
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
63
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
64
  ### Limitations
65
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
66
 
@@ -73,7 +98,7 @@ The SpeechBrain team does not provide any warranty on the performance achieved b
73
  year = {2021},
74
  publisher = {GitHub},
75
  journal = {GitHub repository},
76
- howpublished = {\url{https://github.com/speechbrain/speechbrain}},
77
  }
78
  ```
79
 
 
61
  ### Inference on GPU
62
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
63
 
64
+ ### Training
65
+ The model was trained with SpeechBrain (fc2eabb7).
66
+ To train it from scratch follows these steps:
67
+ 1. Clone SpeechBrain:
68
+ ```bash
69
+ git clone https://github.com/speechbrain/speechbrain/
70
+ ```
71
+ 2. Install it:
72
+ ```
73
+ cd speechbrain
74
+ pip install -r requirements.txt
75
+ pip install -e .
76
+ ```
77
+
78
+ 3. Run Training:
79
+ ```
80
+ cd recipes/WSJ0Mix/separation
81
+ python train.py hparams/sepformer.yaml --data_folder=your_data_folder
82
+ ```
83
+ Note: change num\_spks to 3 in the yaml file.
84
+
85
+ https://drive.google.com/drive/folders/1fcVP52gHgoMX9diNN1JxX_My5KaRNZWs?usp=sharing
86
+
87
+ You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/1ruScDoqiSDNeoDa__u5472UUPKPu54b2?usp=sharing)
88
+
89
  ### Limitations
90
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
91
 
 
98
  year = {2021},
99
  publisher = {GitHub},
100
  journal = {GitHub repository},
101
+ howpublished = {\\url{https://github.com/speechbrain/speechbrain}},
102
  }
103
  ```
104