"""Reproduce the published input/output shape locally; no weights are downloaded."""
import torch
from model import ResidualClassifier

torch.manual_seed(0)
torch.use_deterministic_algorithms(True)
with torch.no_grad():
    model = ResidualClassifier().cpu().eval()
    output = model(torch.ones(1, 64, dtype=torch.float32))
assert list(output.shape) == [1,10]
print("Verified output shape:", list(output.shape))