How to switch on the training pattern of the model in test-step?

[Struggle-Forever]

In some cases, I need to turn on the training mode of the model in the test_step（）to complete the computation, and then turn off the training mode. I have tried using self.train() but it keeps reporting errors.

def test_step():
        self.train()
        do something...
       self.eval()