Bladeren bron

disable pir by default

gaotingquan 1 jaar geleden
bovenliggende
commit
9df73d767e
1 gewijzigde bestanden met toevoegingen van 4 en 1 verwijderingen
  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():