Audio-to-Audio
speechbrain
English
speech-enhancement
PyTorch
speechbrainteam commited on
Commit
1cf44bb
·
1 Parent(s): 9c157fd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -1
README.md CHANGED
@@ -65,6 +65,29 @@ torchaudio.save('enhanced.wav', enhanced.cpu(), 16000)
65
  ### Inference on GPU
66
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
67
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
68
  ### Limitations
69
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
70
 
@@ -92,7 +115,7 @@ title = {SpeechBrain},
92
  year = {2021},
93
  publisher = {GitHub},
94
  journal = {GitHub repository},
95
- howpublished = {\url{https://github.com/speechbrain/speechbrain}},
96
  }
97
  ```
98
 
 
65
  ### Inference on GPU
66
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
67
 
68
+ ### Training
69
+ The model was trained with SpeechBrain (d0accc8).
70
+ To train it from scratch follows these steps:
71
+ 1. Clone SpeechBrain:
72
+ ```bash
73
+ git clone https://github.com/speechbrain/speechbrain/
74
+ ```
75
+ 2. Install it:
76
+ ```
77
+ cd speechbrain
78
+ pip install -r requirements.txt
79
+ pip install -e .
80
+ ```
81
+
82
+ 3. Run Training:
83
+ ```
84
+ cd recipes/Voicebank/enhance/MetricGAN
85
+ python train.py hparams/train.yaml --data_folder=your_data_folder
86
+
87
+ https://drive.google.com/drive/folders/1fcVP52gHgoMX9diNN1JxX_My5KaRNZWs?usp=sharing
88
+
89
+ You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/1fcVP52gHgoMX9diNN1JxX_My5KaRNZWs?usp=sharing)
90
+
91
  ### Limitations
92
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
93
 
 
115
  year = {2021},
116
  publisher = {GitHub},
117
  journal = {GitHub repository},
118
+ howpublished = {\\url{https://github.com/speechbrain/speechbrain}},
119
  }
120
  ```
121