job-gibert.sh 550 Bytes
Newer Older
Gaëtan Caillaut's avatar
Gaëtan Caillaut committed
1
2
3
4
5
6
7
8
9
10
11
12
#!/bin/sh
#SBATCH -N 1
#SBATCH -p gpu
#SBATCH --gres gpu:rtx6000:1
#SBATCH --job-name hatespeech-gibert
#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

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

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