We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent aded85b commit ac91ae3Copy full SHA for ac91ae3
checkpoint_engine/ps.py
@@ -899,6 +899,8 @@ def gather_metas(self, checkpoint_name: str):
899
900
dist.all_gather_object(metas_lst, metas)
901
902
+ self._current_global_parameter_metas = {}
903
+
904
num_parameters = 0
905
all_hosts: list[str] = []
906
global_device_uuids: list[str] = []
0 commit comments