fix: 修复参数传递问题

2026-03-30 22:22:36 +08:00
parent e1f9901384
commit 345fd2f091
6 changed files with 13 additions and 5 deletions
@@ -57,6 +57,8 @@ class Trainer:
            world_size=config.nprocs,
            master_addr=config.master_addr,
            master_port=config.master_port,
+            device_type=config.device_type,
+            device_ids=config.device_ids,
            checkpoint=checkpoint
        )