test.py

import torch

try:
    ckpt = torch.load("out/ckpt.pt", map_location="cpu")
    print("✅ Checkpoint has been loaded successfully")
except Exception as e:
    print("❌ Failed to load the checkpoint:", e)