load_model#

load_model(pretrained: bool = True, path: Optional[Path] = None, device: Union[str, device] = 'cpu') MockModel[source]#

Load MNIST model.

Returns
MnistModel