def METHOD_NAME(self):
    assert self.learner.model is not None
    self.learner.save(self.temp_dir)
    # Make changes to check subsequent load
    self.learner.model = None
    self.learner.batch_size = 42
    self.learner.load(self.temp_dir)
    self.assertIsNotNone(
        self.learner.model, "model is None after loading pth model."
    )
    assert self.learner.batch_size == 2