remove cuda

src/config/config.py

@@ -14,5 +14,5 @@ configuration = {
     "hidden_neurons": 128,
     "dropout_cnn": 0.2,
     "dropout_fc": 0.5,
-    "device": "cuda"
+    "device": "cpu"
 }