From f4c323f537243391e863a014bce85594c7b47374 Mon Sep 17 00:00:00 2001
From: xjqbest <173596896@qq.com>
Date: Thu, 28 May 2020 19:28:59 +0800
Subject: [PATCH] fix

---
 core/trainers/single_infer.py   | 2 +-
 core/trainers/single_trainer.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/core/trainers/single_infer.py b/core/trainers/single_infer.py
index 9028e696..d830fcbb 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 3bf08196..f7239d39 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:
-- 
GitLab