export WANDB_API_KEY=c84b33041138e14a804e9b10d523a7bedac346a4
cd /mnt/data/guibin.chen/open-o1/LLaMA-Factory
source /mnt/data/guibin.chen/open-o1/LLaMA-Factory/venv/bin/activate

export WANDB_PROJECT="re_arc_v3_wlcb"
export WANDB_NAME="qwen_2_re_arc_test"
export FORCE_TORCHRUN=1
# export NNODES=$WORLD_SIZE
# export NPROC_PER_NODE=$KUBERNETES_CONTAINER_RESOURCE_GPU

llamafactory-cli train examples/train_full/qwen2_full_sft_ds3_local_v2_with_pt.yaml > qwen2.log 2>&1