Commit
·
fbd405a
1
Parent(s):
acbaae6
Update README.md
Browse files
README.md
CHANGED
|
@@ -55,6 +55,9 @@ torchaudio.save("source3hat.wav", est_sources[:, :, 2].detach().cpu(), 8000)
|
|
| 55 |
|
| 56 |
```
|
| 57 |
|
|
|
|
|
|
|
|
|
|
| 58 |
#### Referencing SpeechBrain
|
| 59 |
|
| 60 |
```
|
|
@@ -64,7 +67,7 @@ torchaudio.save("source3hat.wav", est_sources[:, :, 2].detach().cpu(), 8000)
|
|
| 64 |
year = {2021},
|
| 65 |
publisher = {GitHub},
|
| 66 |
journal = {GitHub repository},
|
| 67 |
-
howpublished = {
|
| 68 |
}
|
| 69 |
```
|
| 70 |
|
|
|
|
| 55 |
|
| 56 |
```
|
| 57 |
|
| 58 |
+
### Inference on GPU
|
| 59 |
+
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
| 60 |
+
|
| 61 |
#### Referencing SpeechBrain
|
| 62 |
|
| 63 |
```
|
|
|
|
| 67 |
year = {2021},
|
| 68 |
publisher = {GitHub},
|
| 69 |
journal = {GitHub repository},
|
| 70 |
+
howpublished = {\\url{https://github.com/speechbrain/speechbrain}},
|
| 71 |
}
|
| 72 |
```
|
| 73 |
|