| CUDA_VISIBLE_DEVICES=4 python src/lgmodeling/att_matching.py \ | |
| --seed 0 --tgt_len 256 --mem_len 256 --eval_tgt_len 256 \ | |
| --model-a /mnt/data/vinhbk/weights/enwik8/gpt2-finetune/finetune-rope-indice0-heads8-shared1-routed0-topk0-seed0/best_60000 \ | |
| --model-b /mnt/data/vinhbk/weights/enwik8/gpt2-finetune/finetune-rope-indice0-heads8-shared1-routed0-topk0-seed20/best_48000 \ | |
| --data-path /mnt/data/vinhbk/datasets/enwik8 --dataset enwik8 |