Commit
·
95867dd
1
Parent(s):
240f21e
Update README.md
Browse files
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 = {
|
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 |
```
|