diff --git a/core/trainers/single_infer.py b/core/trainers/single_infer.py index 9028e6967aad202e512350ff3e8605383ae95fde..d830fcbbc5f33ce2842e1dd46603797ece0fbae3 100755 --- a/core/trainers/single_infer.py +++ b/core/trainers/single_infer.py @@ -289,7 +289,7 @@ class SingleInfer(TranspileTrainer): def load(self, is_fleet=False): name = "runner." + self._runner_name + "." dirname = envs.get_global_env("epoch.init_model_path", None) - if dirname is None: + if dirname is None or dirname == "": return print("single_infer going to load ", dirname) if is_fleet: diff --git a/core/trainers/single_trainer.py b/core/trainers/single_trainer.py index 3bf08196a6401823a8a10f18a43b8973b3d3b896..f7239d397203cedc58e53df768e6c45a458b8853 100755 --- a/core/trainers/single_trainer.py +++ b/core/trainers/single_trainer.py @@ -288,7 +288,7 @@ class SingleTrainer(TranspileTrainer): def load(self, is_fleet=False): dirname = envs.get_global_env( "runner." + self._runner_name + ".init_model_path", None) - if dirname is None: + if dirname is None or dirname == "": return print("going to load ", dirname) if is_fleet: