File size: 214 Bytes
35d4872 | 1 2 3 4 | 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") |