cemsubakan commited on
Commit
5df8599
·
1 Parent(s): 5568849

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -18
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(), 16000)
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 model was trained with SpeechBrain (fc2eabb7).
65
- To train it from scratch follows these steps:
66
- 1. Clone SpeechBrain:
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/1QiQhp1vi5t4UfNpNETA48_OmPiXnUy8O?usp=sharing).
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.