speechbrainteam
commited on
Commit
•
f8081a9
1
Parent(s):
6cf215c
Update README.md
Browse files
README.md
CHANGED
@@ -52,6 +52,8 @@ torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 16000)
|
|
52 |
|
53 |
|
54 |
```
|
|
|
|
|
55 |
|
56 |
#### Referencing SpeechBrain
|
57 |
|
@@ -62,7 +64,7 @@ torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 16000)
|
|
62 |
year = {2021},
|
63 |
publisher = {GitHub},
|
64 |
journal = {GitHub repository},
|
65 |
-
howpublished = {
|
66 |
}
|
67 |
```
|
68 |
|
|
|
52 |
|
53 |
|
54 |
```
|
55 |
+
### Inference on GPU
|
56 |
+
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
57 |
|
58 |
#### Referencing SpeechBrain
|
59 |
|
|
|
64 |
year = {2021},
|
65 |
publisher = {GitHub},
|
66 |
journal = {GitHub repository},
|
67 |
+
howpublished = {\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
|
68 |
}
|
69 |
```
|
70 |
|