Commit
·
43a1335
1
Parent(s):
279d3f3
Update README.md
Browse files
README.md
CHANGED
@@ -57,12 +57,35 @@ torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
|
|
57 |
### Inference on GPU
|
58 |
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
### Limitations
|
61 |
The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
|
62 |
|
63 |
#### Referencing SpeechBrain
|
64 |
|
65 |
-
|
66 |
```bibtex
|
67 |
@misc{speechbrain,
|
68 |
title={SpeechBrain: A General-Purpose Speech Toolkit},
|
@@ -73,7 +96,6 @@ The SpeechBrain team does not provide any warranty on the performance achieved b
|
|
73 |
primaryClass={eess.AS}
|
74 |
}
|
75 |
```
|
76 |
-
```
|
77 |
|
78 |
#### Referencing SepFormer
|
79 |
```bibtex
|
|
|
57 |
### Inference on GPU
|
58 |
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
59 |
|
60 |
+
### Training
|
61 |
+
The model was trained with SpeechBrain (e375cd13).
|
62 |
+
To train it from scratch follows these steps:
|
63 |
+
1. Clone SpeechBrain:
|
64 |
+
```bash
|
65 |
+
git clone https://github.com/speechbrain/speechbrain/
|
66 |
+
```
|
67 |
+
2. Install it:
|
68 |
+
```
|
69 |
+
cd speechbrain
|
70 |
+
pip install -r requirements.txt
|
71 |
+
pip install -e .
|
72 |
+
```
|
73 |
+
|
74 |
+
3. Run Training:
|
75 |
+
```
|
76 |
+
cd recipes/WHAMandWHAMR/separation
|
77 |
+
python train.py hparams/sepformer-wham.yaml --data_folder=your_data_folder
|
78 |
+
```
|
79 |
+
|
80 |
+
You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/1dIAT8hZxvdJPZNUb8Zkk3BuN7GZ9-mZb?usp=sharing).
|
81 |
+
|
82 |
+
|
83 |
### Limitations
|
84 |
The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
|
85 |
|
86 |
#### Referencing SpeechBrain
|
87 |
|
88 |
+
|
89 |
```bibtex
|
90 |
@misc{speechbrain,
|
91 |
title={SpeechBrain: A General-Purpose Speech Toolkit},
|
|
|
96 |
primaryClass={eess.AS}
|
97 |
}
|
98 |
```
|
|
|
99 |
|
100 |
#### Referencing SepFormer
|
101 |
```bibtex
|