-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathoof-bert-uncased-k-fold.sh
9 lines (7 loc) · 1.02 KB
/
oof-bert-uncased-k-fold.sh
1
2
3
4
5
6
7
8
9
# oof for 10 fold bert-base-uncased, question_answer, seed 2020
python oof-k-fold.py --model_type "bert" --model_name "bert-base-uncased" --content "Question_Answer" --seed 2020 --n_splits 10 --split "GroupKfold" --valid_batch_size 32 --loss "bce" --augment --swa --num_workers 4
# oof for 5 fold bert-base-uncased, question + answer, seed 2020
python oof-k-fold.py --model_type "bert" --model_name "bert-base-uncased" --content "Question" --seed 2020 --n_splits 5 --split "GroupKfold" --valid_batch_size 32 --loss "bce" --augment --swa --num_workers 4
python oof-k-fold.py --model_type "bert" --model_name "bert-base-uncased" --content "Answer" --seed 2020 --n_splits 5 --split "GroupKfold" --valid_batch_size 32 --loss "bce" --augment --swa --num_workers 4 --merge
# oof for 5 fold bert-base-uncased, question_answer, seed 1010
# python oof-k-fold.py --model_type "bert" --model_name "bert-base-uncased" --content "Question_Answer" --seed 1010 --n_splits 5 --split "GroupKfold" --valid_batch_size 32 --loss "bce" --augment --swa --num_workers 4