chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id
- torch 2.7.1-cu126 升级至 2.11.0-cu128,numpy 2.3.2 升级至 2.4.4 - 移除 astrai/trainer/optim.py,改用 torch.optim.Muon - parallel setup: gloo 后端不再传递 device_id,单卡多进程不再报错
This commit is contained in:
@@ -47,7 +47,6 @@ from astrai.trainer import (
|
||||
BaseScheduler,
|
||||
BaseStrategy,
|
||||
CallbackFactory,
|
||||
Muon,
|
||||
SchedulerFactory,
|
||||
StrategyFactory,
|
||||
TrainCallback,
|
||||
@@ -75,7 +74,6 @@ __all__ = [
|
||||
"GenerationRequest",
|
||||
"InferenceEngine",
|
||||
"LoRAConfig",
|
||||
"Muon",
|
||||
"Pipeline",
|
||||
"PipelineConfig",
|
||||
"ProtocolHandler",
|
||||
|
||||
Reference in New Issue
Block a user