Explorar o código

disable pir by default

gaotingquan hai 1 ano
pai
achega
9df73d767e
Modificáronse 1 ficheiros con 4 adicións e 1 borrados
  1. 4 1
      paddlex/engine.py

+ 4 - 1
paddlex/engine.py

@@ -14,7 +14,7 @@
 
 import os
 
-
+from .utils.lazy_loader import disable_pir_bydefault
 from .utils.result_saver import try_except_decorator
 from .utils.config import parse_args, get_config
 from .utils.errors import raise_unsupported_api_error
@@ -37,10 +37,13 @@ class Engine(object):
         if self._mode == "check_dataset":
             return self._model.check_dataset()
         elif self._mode == "train":
+            disable_pir_bydefault()
             self._model.train()
         elif self._mode == "evaluate":
+            disable_pir_bydefault()
             return self._model.evaluate()
         elif self._mode == "export":
+            disable_pir_bydefault()
             return self._model.export()
         elif self._mode == "predict":
             for res in self._model.predict():