File size: 1,245 Bytes
a2ffd07 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | #!/bin/bash
#SBATCH --job-name=vis_hal_circuit # Job name
#SBATCH --output=./hallucination/scripts/out.txt # Output file
#SBATCH --error=./hallucination/scripts/error.txt # Error file
#SBATCH --ntasks=1 # Number of tasks (processes)
#SBATCH --gpus=1 # Number of GPUs per node
#SBATCH --nodes=1 # Số node yêu cầu
#SBATCH --cpus-per-task=20 # Số CPU cho mỗi task
cd ..
declare -a text_sae_types=("crossattn")
declare -a vis_sae_types=("post")
python run_circuit_with_plot.py \
--model_name "Salesforce/blip-image-captioning-base" \
--tok_name "Salesforce/blip-image-captioning-base" \
--data_path "./COCO-Dataset/filtered_val/hallucinated/75719" \
--target_position -5 \
--batch_size 16 \
--text_sae_types "${text_sae_types[@]}" \
--vis_sae_types "${vis_sae_types[@]}" \
--text_batch 1000 \
--vis_batch 200 \
--crop_ratio 1.0 \
--inter True \
--intra True \
--gradient_mode "ig" \
--edge_gradient_mode "gradient" \
--num_nodes 50 \
--scoring_mode "less" \
--ig_steps 10 \
--filter_seq_length 30 \
--save_name "vis_file/sae_graph_75719_nonhal.html" \
--dtype "bfloat16" \
--device_id 0 |