From 9994eed790c53eb1811b674448ae203afd463bcb Mon Sep 17 00:00:00 2001
From: Nicolas Farrugia <nicolas.farrugia@imt-atlantique.fr>
Date: Fri, 1 Apr 2022 14:21:04 +0200
Subject: [PATCH] added gpu

---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index 0d3b9e2..0431228 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. 
-- 
GitLab