Commit f7b3e791 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

switch device to cuda

parent 1dd132be
......@@ -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
......@@ -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
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment