load_model

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

Load MNIST model.

Returns

MnistModel

previous

load_dataset

next

MnistModel