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")