# CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
#     --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads12-shared1-routed0-topk0-seed0/best_65052 \
#     --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads12-shared1-routed0-topk0-seed20/best_5004 \
#     --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 
# CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
#     --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads12-shared1-routed0-topk0-seed0/best_65052 \
#     --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads12-shared1-routed0-topk0-seed40/best_5004 \
#     --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 
# CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
#     --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads12-shared1-routed0-topk0-seed20/best_5004 \
#     --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads12-shared1-routed0-topk0-seed40/best_5004 \
#     --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 


CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
    --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads8-shared1-routed0-topk0-seed0/best_10008 \
    --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads8-shared1-routed0-topk0-seed20/best_85068 \
    --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 
CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
    --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads8-shared1-routed0-topk0-seed0/best_10008 \
    --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads8-shared1-routed0-topk0-seed40/best_35028  \
    --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 
CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
    --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads8-shared1-routed0-topk0-seed20/best_85068 \
    --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads8-shared1-routed0-topk0-seed40/best_35028 \
    --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 


CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
    --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads16-shared1-routed0-topk0-seed0/best_5004 \
    --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads16-shared1-routed0-topk0-seed20/best_5004 \
    --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 
CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
    --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads16-shared1-routed0-topk0-seed0/best_5004 \
    --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads16-shared1-routed0-topk0-seed40/best_5004 \
    --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet 
CUDA_VISIBLE_DEVICES=7 python src/imagenet/att_matching.py \
    --model-a /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads16-shared1-routed0-topk0-seed20/best_5004 \
    --model-b /mnt/data/vinhbk/weights/imagenet/vit-finetune/finetune-learnable-indice0-heads16-shared1-routed0-topk0-seed40/best_5004 \
    --data-path /mnt/data/vinhbk/datasets/imagenet --save-path plots/imagenet