file_name="0607ce86"
LOOP_CKPT="saves/loop_runs/2025-11-28/checkpoint_final.pt"

python test_time_train_ARC.py \
          --epochs 10 \
          --batch-size 8 \
          --image-size 64 \
          --patch-size 2 \
          --learning-rate 3e-4 \
          --weight-decay 0 \
          --embed-dim 512 \
          --mlp-dim 1024 \
          --num-heads 8 \
          --num-colors 12 \
          --loop-core-depth 2 \
          --max-loop-steps 6 \
          --min-loop-steps 6 \
          --resume-checkpoint "${LOOP_CKPT}" \
          --lr-scheduler "cosine" \
          --train-split "eval_color_permute_ttt_9/${file_name}" \
          --data-root "raw_data/ARC-AGI" \
          --eval-split "eval_color_permute_ttt_9/${file_name}" \
          --resume-skip-task-token \
          --architecture "loop_vit" \
          --eval-save-name "ARC_1_eval_LoopViT_sanity" \
          --num-attempts 5 \
          --ttt-num-each 1