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-base \
--task_name mnli \
--work_dir checkpoint/roberta-base \
--max_seq_length 512 \
--train_batch_size 64 \
--eval_batch_size 64 \
--weight_decay 0.12 \
--arch_weight_decay 0.1 \
--lr 2e-5 \
--arch_lr 3e-5 \
--apply_lora \
--lora_r 8 \
--lora_alpha 8 \
--seed 0 \
--valid_step 100 \
--unroll_step 1 \
--warmup_step 600 \
--max_step 10000 \
--train_iters 10000 \
--save_step 9800 \

