revert to compile train_step only due to pt2 issue

open-mmlab · Mar 9, 2023 · fdb57b7 · fdb57b7
1 parent 20242e1
commit fdb57b7
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/mmengine/runner/runner.py b/mmengine/runner/runner.py
@@ -1718,10 +1718,6 @@ def val(self) -> dict:
         # make sure checkpoint-related hooks are triggered after `before_run`
         self.load_or_resume()
 
-        # Maybe compile the model according to options in self.cfg.compile
-        # This must be called **AFTER** model has been wrapped.
-        self._maybe_compile('val_step')
-
         metrics = self.val_loop.run()  # type: ignore
         self.call_hook('after_run')
         return metrics
@@ -1745,10 +1741,6 @@ def test(self) -> dict:
         # make sure checkpoint-related hooks are triggered after `before_run`
         self.load_or_resume()
 
-        # Maybe compile the model according to options in self.cfg.compile
-        # This must be called **AFTER** model has been wrapped.
-        self._maybe_compile('test_step')
-
         metrics = self.test_loop.run()  # type: ignore
         self.call_hook('after_run')
         return metrics