speechbrainteam commited on
Commit
956c1e4
1 Parent(s): 8bad015

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -67,6 +67,8 @@ enhanced = enhance_model.enhance_file("speechbrain/mtl-mimic-voicebank/example.w
67
  # Saving enhanced signal on disk
68
  torchaudio.save('enhanced.wav', enhanced.unsqueeze(0).cpu(), 16000)
69
  ```
 
 
70
 
71
  ## Referencing Mimic Loss
72
 
@@ -92,6 +94,6 @@ title = {SpeechBrain},
92
  year = {2021},
93
  publisher = {GitHub},
94
  journal = {GitHub repository},
95
- howpublished = {\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
96
  }
97
  ```
 
67
  # Saving enhanced signal on disk
68
  torchaudio.save('enhanced.wav', enhanced.unsqueeze(0).cpu(), 16000)
69
  ```
70
+ ### Inference on GPU
71
+ To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
72
 
73
  ## Referencing Mimic Loss
74
 
 
94
  year = {2021},
95
  publisher = {GitHub},
96
  journal = {GitHub repository},
97
+ howpublished = {\\\\\\\\\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
98
  }
99
  ```