| 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"], | |
| ) | |