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 ...@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-davidson-dataset.csv
--output output/roberta/davidson \ --output output/roberta/davidson \
--epochs 50 \ --epochs 50 \
--optimizer adam \ --optimizer adam \
--device cpu \ --device cuda \
--jobname roberta/davidson \ --jobname roberta/davidson \
--folds 5 --folds 5
...@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-gibert-dataset.csv \ ...@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-gibert-dataset.csv \
--output output/roberta/gibert \ --output output/roberta/gibert \
--epochs 50 \ --epochs 50 \
--optimizer adam \ --optimizer adam \
--device cpu \ --device cuda \
--jobname roberta/gibert \ --jobname roberta/gibert \
--folds 5 --folds 5
...@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-hate-speech-dataset.c ...@@ -17,6 +17,6 @@ python train_roberta.py train-roberta --input data/cleaned-hate-speech-dataset.c
--output output/roberta/merged \ --output output/roberta/merged \
--epochs 50 \ --epochs 50 \
--optimizer adam \ --optimizer adam \
--device cpu \ --device cuda \
--jobname roberta/merged \ --jobname roberta/merged \
--folds 5 --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