Commit
·
5df8599
1
Parent(s):
5568849
Update README.md
Browse files
README.md
CHANGED
@@ -53,7 +53,7 @@ model = separator.from_hparams(source="speechbrain/sepformer-whamr-enhancement",
|
|
53 |
# for custom file, change path
|
54 |
est_sources = model.separate_file(path='speechbrain/sepformer-whamr-enhancement/example_whamr.wav')
|
55 |
|
56 |
-
torchaudio.save("enhanced_whamr.wav", est_sources[:, :, 0].detach().cpu(),
|
57 |
|
58 |
```
|
59 |
|
@@ -61,26 +61,13 @@ torchaudio.save("enhanced_whamr.wav", est_sources[:, :, 0].detach().cpu(), 16000
|
|
61 |
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
62 |
|
63 |
### Training
|
64 |
-
The
|
65 |
-
|
66 |
-
|
67 |
-
```bash
|
68 |
-
git clone https://github.com/speechbrain/speechbrain/
|
69 |
-
```
|
70 |
-
2. Install it:
|
71 |
-
```
|
72 |
-
cd speechbrain
|
73 |
-
pip install -r requirements.txt
|
74 |
-
pip install -e .
|
75 |
-
```
|
76 |
|
77 |
-
3. Run Training:
|
78 |
-
```
|
79 |
-
cd recipes/WHAMandWHAMR/separation/
|
80 |
-
python train.py hparams/sepformer-whamr.yaml --data_folder=your_data_folder --sample_rate=16000
|
81 |
```
|
82 |
|
83 |
-
You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/
|
84 |
|
85 |
### Limitations
|
86 |
The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
|
|
|
53 |
# for custom file, change path
|
54 |
est_sources = model.separate_file(path='speechbrain/sepformer-whamr-enhancement/example_whamr.wav')
|
55 |
|
56 |
+
torchaudio.save("enhanced_whamr.wav", est_sources[:, :, 0].detach().cpu(), 8000)
|
57 |
|
58 |
```
|
59 |
|
|
|
61 |
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
62 |
|
63 |
### Training
|
64 |
+
The training script is currently being worked on an ongoing pull-request.
|
65 |
+
|
66 |
+
We will update the model card as soon as the PR is merged.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
67 |
|
|
|
|
|
|
|
|
|
68 |
```
|
69 |
|
70 |
+
You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/1V0KwkEfWwomZ0Vjox0BTnQ694_uxgu8G).
|
71 |
|
72 |
### Limitations
|
73 |
The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
|