We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2a36e0e commit 5cb9ebeCopy full SHA for 5cb9ebe
trinity/explorer/explorer.py
@@ -121,7 +121,7 @@ def _update_model_weight(self, state_dict: dict) -> None:
121
self.state_dict = state_dict
122
update_weight_args_list = []
123
for name, param in state_dict.items():
124
- update_weight_args_list.append((name, param.dtype, param.shape))
+ update_weight_args_list.append((name, str(param.dtype), tuple(param.shape)))
125
ray.get([model.sync_model.remote(update_weight_args_list) for model in self.models])
126
self.state_dict.clear()
127
0 commit comments