mnist_classifier/load.py

import pickle

from classifier import Classifier

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

model = Classifier(
    num_classes=meta["num_classes"],
    base_channels=meta["base_channels"],
)