fix opt-offload save bug (#36433)

e703a2ed · duanboqiang · GitHub · 12882b2f · e703a2ed
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py ...addle/distributed/fleet/meta_optimizers/sharding/utils.py +1 -1

未找到文件。
--- a/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py
+++ b/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py
@@ -906,7 +906,7 @@ def save_persistables(exe, dirname, main_program, filename=None):
            "_velocity_0"
        ]
        for check in checks:
-            if var.name.endswith(check):
+            if var.name.endswith(check) and var.persistable:
                return True
        return False