Skip to content

Commit 5cb9ebe

Browse files
committed
fix checkpoint sync mode
1 parent 2a36e0e commit 5cb9ebe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

trinity/explorer/explorer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def _update_model_weight(self, state_dict: dict) -> None:
121121
self.state_dict = state_dict
122122
update_weight_args_list = []
123123
for name, param in state_dict.items():
124-
update_weight_args_list.append((name, param.dtype, param.shape))
124+
update_weight_args_list.append((name, str(param.dtype), tuple(param.shape)))
125125
ray.get([model.sync_model.remote(update_weight_args_list) for model in self.models])
126126
self.state_dict.clear()
127127

0 commit comments

Comments
 (0)