fix no test_step error in TorchWrapper (#1600)

Author: AllentDan

tools/profiler.py

@@ -80,8 +80,8 @@ def __init__(self, model):
         self.model = model
 
     @TimeCounter.count_time(Backend.PYTORCH.value)
-    def forward(self, *args, **kwargs):
-        return self.model(*args, **kwargs)
+    def test_step(self, *args, **kwargs):
+        return self.model.test_step(*args, **kwargs)
 
 
 def main():