LTA_openwebtext_dualt/scripts/eval_ar_8gpu_step20k_20260507.py

#!/usr/bin/env python3
from __future__ import annotations

import importlib.util
import sys
from pathlib import Path


BASE = Path(__file__).with_name("eval_lm1b_latest_non_owt_methods_genppl_20260506.py")
spec = importlib.util.spec_from_file_location("eval_lm1b_latest_non_owt_methods_genppl_20260506", BASE)
if spec is None or spec.loader is None:
    raise RuntimeError(f"Could not import {BASE}")
mod = importlib.util.module_from_spec(spec)
sys.modules[spec.name] = mod
spec.loader.exec_module(mod)

mod.base.TARGETS = [
    mod.base.EvalTarget(
        "ar_8gpu_scratch_step20k",
        "ar",
        "runs/ar_lm1b_flmpack_bert_small_len128_gbs512_8gpu_1m_scratch_20260505/step_0020000.pt",
    )
]
mod.TARGETS = mod.base.TARGETS

if __name__ == "__main__":
    mod.base.main()