pere commited on
Commit
3bd5df5
·
1 Parent(s): 03914c5
run_sup_example.sh → run_sup.sh RENAMED
File without changes
run_sup_roberta.sh ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ export ZONE=us-central2-b
4
+ export XRT_TPU_CONFIG="localservice;0;localhost:51011"
5
+ export TPU_NUM_DEVICES=4
6
+ export ALLOW_MULTIPLE_LIBTPU_LOAD=1
7
+
8
+ python ../../SimCSE/train.py \
9
+ --model_name_or_path pere/roberta-base-scand-xlm \
10
+ --train_file data/mnli_no_for_simcse.csv \
11
+ --output_dir result/sup-simcse-pere-roberta-scand-base \
12
+ --num_train_epochs 3 \
13
+ --per_device_train_batch_size 128 \
14
+ --learning_rate 5e-5 \
15
+ --max_seq_length 32 \
16
+ --evaluation_strategy steps \
17
+ --metric_for_best_model stsb_spearman \
18
+ --load_best_model_at_end \
19
+ --eval_steps 125 \
20
+ --pooler_type cls \
21
+ --overwrite_output_dir \
22
+ --temp 0.05 \
23
+ --do_train \
24
+ --do_eval \
25
+ "$@"
run_unsup_example.sh → run_unsup.sh RENAMED
File without changes