diff --git a/configs/basic_model.json b/configs/basic_model.json
index 55d9af8..55fac2a 100644
--- a/configs/basic_model.json
+++ b/configs/basic_model.json
@@ -48,10 +48,10 @@
   },
   "trainer": {
     "optimizer": {
-      "type": "adam",
+      "type": "adagrad",
       "lr": 0.001
     },
-    "num_epochs": 2,
-    "cuda_device": 1
+    "num_epochs": 20,
+    "cuda_device": 3
   }
 }
diff --git a/testing/intent_predictor.py b/testing/intent_predictor.py
index 16d2603..d1541d3 100644
--- a/testing/intent_predictor.py
+++ b/testing/intent_predictor.py
@@ -51,8 +51,8 @@ def make_predictions(model: Model, dataset_reader: DatasetReader, file_path: str
 
 def load_model_and_run_predictions(saved_model_dir: str):
     project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-    train_file_path = project_root + '/data/tsv/train.tsv'
-    test_file_path = project_root + '/data/tsv/test.tsv'
+    dev_file_path = project_root + '/data/jsonl/dev.jsonl'
+    test_file_path = project_root + '/data/jsonl/test.jsonl'
 
     model_archive = load_archive(os.path.join(saved_model_dir, 'model.tar.gz'))
     citation_dataset_reader = CitationDataSetReader()