#!/bin/bash
# this is a demo script to run the baseline method FASP
# Make sure to set the correct paths for your output directories and modify the parameters as needed.

python main_FPVE.py \
    --model gpt2 \
    --method FASP \
    --pruned_heads_ratio 0.1 \
    --seed 1 \
    --output_dir YOUR_OUTPUT_PATH \
    --device "cuda:0" \
    --gamma 0.3 \
    --scalar 1 \
    --fitness_mode 'BIAS_PPL'