python src/lgmodeling/plot.py --output-dir plots/enwik8/learnable-indice0-heads4-shared1-routed0-topk0.pdf\
     --file-1  results/enwik8/[finetune-learnable-indice0-heads4-shared1-routed0-topk0-seed0+finetune-learnable-indice0-heads4-shared1-routed0-topk0-seed20].json\
     --file-2  results/enwik8/[finetune-learnable-indice0-heads4-shared1-routed0-topk0-seed0+finetune-learnable-indice0-heads4-shared1-routed0-topk0-seed40].json\
     --file-3  results/enwik8/[finetune-learnable-indice0-heads4-shared1-routed0-topk0-seed20+finetune-learnable-indice0-heads4-shared1-routed0-topk0-seed40].json