bug fix

b3a4f946 · yaoxuefeng · fec24e9f · b3a4f946
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

core/trainers/single_trainer.py core/trainers/single_trainer.py +1 -1

未找到文件。
--- a/core/trainers/single_trainer.py
+++ b/core/trainers/single_trainer.py
@@ -328,7 +328,7 @@ class SingleTrainer(TranspileTrainer):
        if is_fleet:
            fleet.load_persistables(self._exe, dirname)
        else:
-            if load_vars is None or dirname == "":
+            if load_vars is None or len(load_vars) == 0:
                fluid.io.load_persistables(self._exe, dirname)
            else:
                fluid.io.load_vars(