MnistModel.set\_extra\_state ============================ .. currentmodule:: deepchecks.vision.datasets.classification.mnist_torch .. automethod:: MnistModel.set_extra_state