diff --git a/drugex/training/generators/sequence_rnn.py b/drugex/training/generators/sequence_rnn.py
index 8583d73..558a6c6 100644
--- a/drugex/training/generators/sequence_rnn.py
+++ b/drugex/training/generators/sequence_rnn.py
@@ -43,7 +43,10 @@ def attachToGPUs(self, gpus):
         -------
         None
         """
-        self.device = torch.device(f'cuda:{gpus[0]}')
+        if gpus[0] == -1:
+            self.device = torch.device('cpu')
+        else:
+            self.device = torch.device(f'cuda:{gpus[0]}')
         self.to(self.device)
         self.gpus = (gpus[0],)