lmc-code / scripts /cifar100 /attn_match.sh
khanhvinh9's picture
Upload folder using huggingface_hub
a20151e verified
Raw
History Blame Contribute Delete
929 Bytes
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