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

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