cifar10_classifier/load.py

import pickle

from classifier import Cifar10CNN

with open("cifar10_classifier.pkl", "rb") as f:
    bundle = pickle.load(f)
meta = bundle["meta"]
params = bundle["params"]

model = Cifar10CNN(
    num_classes=meta["num_classes"],
    channels=tuple(meta["channels"]),
    groups=meta["groups"],
)