cxr-vlm-code / opti.py
convitom
f
35d4872
raw
history blame contribute delete
214 Bytes
import torch
ckpt = torch.load(r"C:\Users\admin\Downloads\optimizer.pt", map_location="cpu", weights_only=False)
for i, g in enumerate(ckpt["param_groups"]):
print(f"saved group {i}: {len(g['params'])} params")