speechbrainteam commited on
Commit
f8081a9
1 Parent(s): 6cf215c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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 = {\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
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