job-davidson.sh 558 Bytes
Newer Older
1
#!/bin/sh
Gaëtan Caillaut's avatar
Gaëtan Caillaut committed
2
3
4
5
6
7
8
9
10
11
#SBATCH -N 1
#SBATCH -p gpu
#SBATCH --gres gpu:rtx6000:1
#SBATCH --job-name hatespeech-davidson
#SBATCH --time 3-0
#SBATCH --mem 20G
#SBATCH -o logs/out-%j.txt
#SBATCH -e logs/err-%j.txt
#SBATCH --mail-type=ALL
#SBATCH --mail-user=gaetan.caillaut@univ-lemans.fr
12

Gaëtan Caillaut's avatar
Gaëtan Caillaut committed
13
14
eval "$(conda shell.bash hook)"
conda activate polysemy
15

Gaëtan Caillaut's avatar
Gaëtan Caillaut committed
16
python train_roberta.py train-roberta --input data/cleaned-davidson-dataset.csv \
17
18
19
20
    --output output/roberta/davidson \
    --epochs 50 \
    --optimizer adam \
    --device cpu \
Gaëtan Caillaut's avatar
Gaëtan Caillaut committed
21
    --jobname roberta/davidson \
22
    --folds 5