Skip to content

Commit 68a06ce

Browse files
[Train] Fix device_type attr bug (flagos-ai#1040)
### PR Category Train ### PR Types Bug Fixes ### PR Description Fix device_type attr bug in hetero train with n nodes.
1 parent da4d68f commit 68a06ce

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

flagscale/runner/launcher/launcher_ssh.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -212,8 +212,8 @@ def _run_each(
212212
node_specific_config = {}
213213
if device_type is not None:
214214
node_specific_config = (
215-
self.device_type_specific.get(device_type, {})
216-
if self.device_type_specific
215+
self.backend.device_type_specific.get(device_type, {})
216+
if self.backend.device_type_specific
217217
else {}
218218
)
219219
node_specific_config.update(

0 commit comments

Comments
 (0)