Audio-to-Audio
speechbrain
English
speech-enhancement
PyTorch
speechbrainteam commited on
Commit
95867dd
·
1 Parent(s): 240f21e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -59,6 +59,9 @@ enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
59
  torchaudio.save('enhanced.wav', enhanced.cpu(), 16000)
60
  ```
61
 
 
 
 
62
  ## Referencing MetricGAN+
63
 
64
  If you find MetricGAN+ useful, please cite:
@@ -83,6 +86,6 @@ title = {SpeechBrain},
83
  year = {2021},
84
  publisher = {GitHub},
85
  journal = {GitHub repository},
86
- howpublished = {\\\\url{https://github.com/speechbrain/speechbrain}},
87
  }
88
  ```
 
59
  torchaudio.save('enhanced.wav', enhanced.cpu(), 16000)
60
  ```
61
 
62
+ ### Inference on GPU
63
+ To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
64
+
65
  ## Referencing MetricGAN+
66
 
67
  If you find MetricGAN+ useful, please cite:
 
86
  year = {2021},
87
  publisher = {GitHub},
88
  journal = {GitHub repository},
89
+ howpublished = {\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
90
  }
91
  ```