python run_classification.py \
--model="gpt2-xl" \
--dataset="sst2" \
--num_seeds=5 \
--start_seed=0 \
--all_shots="0,1,4,8" \
--bs=2 \
--method='gmm_train_estimate' \
--gmm_train_estimate_scale=300

