Quillan-Ronin / scripts /check_ckpt.py
CrashOverrideX's picture
Upload batch 15/16
7500ff7 verified
Raw History Blame Contribute Delete
326 Bytes
import torch
from pathlib import Path
for d in [r"C:\02_QUILLAN\checkpoints\checkpoints_oni"]:
for p in Path(d).glob("*.pt"):
ck = torch.load(str(p), map_location="cpu", weights_only=False)
print(p.name, "step", ck.get("step"), "best_val", ck.get("best_val"))
print(" keys:", list(ck.keys())[:5])