export num_gpus=1
export CUBLAS_WORKSPACE_CONFIG=":16:8" # https://docs.nvidia.com/cuda/cublas/index.html#cublasApi_reproducibility
export CUDA_VISIBLE_DEVICES=0
python examples/text-classification/run_glue_bilevel.py \
--model_name_or_path roberta-large \
--task_name stsb \
--model_init_checkpoint checkpoint/roberta-large/mnli/model.13900.pt \
--arch_init_checkpoint checkpoint/roberta-large/mnli/arch.13900.pt \
--max_seq_length 128 \
--train_batch_size 32 \
--eval_batch_size 128 \
--weight_decay 0.1 \
--lr 1e-5 \
--arch_lr 5e-6 \
--arch_weight_decay 0.1 \
--apply_lora \
--lora_r 8 \
--lora_alpha 16 \
--seed 0 \
--valid_step 10 \
--unroll_step 1 \
--warmup_step 120 \
--max_step 2000 \
--train_iters 2000 \
