update run scripts

run_sup_example.sh

@@ -1,3 +1,10 @@
+#!/bin/bash
+
+export ZONE=us-central2-b
+export XRT_TPU_CONFIG="localservice;0;localhost:51011"
+export TPU_NUM_DEVICES=4
+export ALLOW_MULTIPLE_LIBTPU_LOAD=1
+
 python ../../SimCSE/train.py \
     --model_name_or_path NbAiLab/nb-bert-base \
     --train_file data/mnli_no_for_simcse.csv \

run_unsup_example.sh

@@ -3,6 +3,10 @@
 # In this example, we show how to train SimCSE on unsupervised Wikipedia data.
 # If you want to train it with multiple GPU cards, see "run_sup_example.sh"
 # about how to use PyTorch's distributed data parallel.
+export ZONE=us-central2-b
+export XRT_TPU_CONFIG="localservice;0;localhost:51011"
+export TPU_NUM_DEVICES=4
+export ALLOW_MULTIPLE_LIBTPU_LOAD=1
 
 python3 ../../SimCSE/train.py \
     --model_name_or_path NbAiLab/nb-bert-base \