fix opt-offload save bug (#36433)

PaddlePaddle · Oct 15, 2021 · e703a2e · e703a2e
1 parent 12882b2
commit e703a2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 is_opt_vars(var):
             "_velocity_0"
         ]
         for check in checks:
-            if var.name.endswith(check):
+            if var.name.endswith(check) and var.persistable:
                 return True
         return False