diff --git a/README.md b/README.md index 0d3b9e2c598da219c49ea3ffd0ede0265545cc7c..0431228eb7c401c1fbf3d29a815ca83fd331f13d 100644 --- a/README.md +++ b/README.md @@ -19,6 +19,8 @@ The code should pass validation by the [Hear validator](https://github.com/neura `hear-validator soundnetbrain_hear --model /path/to/modefile.pt -d cpu` +It also works on GPU now +`hear-validator soundnetbrain_hear --model /path/to/modefile.pt -d cuda` ## Running the full evaluation Install the [heareval package](https://github.com/neuralaudio/hear-eval-kit). Can be done with `pip install heareval`, but it might be more appropriate to use docker.