def METHOD_NAME(self):
    self.learner.ort_session = None
    self.learner.load(self.temp_dir / "weights" / f"x3d_{_BACKBONE}.pyth")
    self.learner.optimize()
    assert self.learner._ort_session is not None
    step_input = self.learner._example_input.repeat(
        self.learner.batch_size, 1, 1, 1
    )
    step_output = self.learner.infer(step_input)
    assert isinstance(step_output[0], Category)
    # Clean up
    self.learner.ort_session = None