Skip to content

Commit 279a908

Browse files
authored
bugfix: reset global meta when gather meta (#57)
1 parent aded85b commit 279a908

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

checkpoint_engine/ps.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -899,6 +899,8 @@ def gather_metas(self, checkpoint_name: str):
899899

900900
dist.all_gather_object(metas_lst, metas)
901901

902+
self._current_global_parameter_metas = {}
903+
902904
num_parameters = 0
903905
all_hosts: list[str] = []
904906
global_device_uuids: list[str] = []

0 commit comments

Comments
 (0)