job-merged.sh 337 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

export PYTHONPATH="$HOME/polysemy/minibert:$PYTHONPATH"

python train_minibert.py train-minibert --input data/cleaned-hate-speech-dataset.csv \
    --output output/minibert/merged \
    --epochs 50 \
    --optimizer adam \
    --device cpu \
    --jobname minibert/merged \
    --tokenizer data/tokenizer.json \
    --folds 5