CUDA_VISIBLE_DEVICES=0 python src/imagenet/att_matching.py --input-size 32 --data-set CIFAR10\
    --model-a /mnt/d/NCKH/weights/lmc/cifar10/vit-finetune/finetune0.005-learnable-indice0-heads4-shared1-routed0-topk0-mlpFalse-seed20/best_19110 \
    --model-b /mnt/d/NCKH/weights/lmc/cifar10/vit-finetune/finetune0.005-learnable-indice0-heads4-shared1-routed0-topk0-mlpFalse-seed40/best_16380 \
    --data-path /mnt/d/NCKH/datasets/cifar10 --save-path plots/cifar10 

CUDA_VISIBLE_DEVICES=0 python src/imagenet/att_matching.py --input-size 32 --data-set CIFAR10\
    --model-a /mnt/d/NCKH/weights/lmc/cifar10/vit-finetune/finetune0.005-learnable-indice0-heads4-shared1-routed0-topk0-mlpFalse-seed0/best_16380 \
    --model-b /mnt/d/NCKH/weights/lmc/cifar10/vit-finetune/finetune0.005-learnable-indice0-heads4-shared1-routed0-topk0-mlpFalse-seed40/best_16380 \
    --data-path /mnt/d/NCKH/datasets/cifar10 --save-path plots/cifar10