refactor: 优化并行训练配置与启动管理
- 配置新增 start_method 支持 spawn/fork/forkserver 选择 - 启动方式 mp.spawn 改为 mp.start_processes,支持 daemon=True - validate() 改为基于 metadata 的反射式校验,不再硬编码字段列表 - CLI 新增 --start_method 参数
This commit is contained in:
@@ -52,6 +52,7 @@ class Trainer:
|
||||
master_addr=cfg.master_addr,
|
||||
master_port=cfg.master_port,
|
||||
device_type=cfg.device_type,
|
||||
start_method=cfg.start_method,
|
||||
checkpoint=checkpoint,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user