switch device to cuda

jobs/roberta/job-davidson.sh

@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-davidson-dataset.csv
     --output output/roberta/davidson \
     --epochs 50 \
     --optimizer adam \
-    --device cpu \
+    --device cuda \
     --jobname roberta/davidson \
     --folds 5

jobs/roberta/job-gibert.sh

@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-gibert-dataset.csv \
     --output output/roberta/gibert \
     --epochs 50 \
     --optimizer adam \
-    --device cpu \
+    --device cuda \
     --jobname roberta/gibert \
     --folds 5

jobs/roberta/job-merged.sh

@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-hate-speech-dataset.c
     --output output/roberta/merged \
     --epochs 50 \
     --optimizer adam \
-    --device cpu \
+    --device cuda \
     --jobname roberta/merged \
     --folds 5