refactor: remove redundant strategy/executor code
- Drop BaseStrategy.model_fn (stored but never read) - Drop model_fn= passed to StrategyFactory.create in train_context - Simplify FSDPExecutor.clip_grad_norm None branch to delegate to super() - Remove DDPExecutor._gather_state_dict override (identical to base)
This commit is contained in:
@@ -211,7 +211,6 @@ class TrainContextBuilder:
|
||||
model=context.model,
|
||||
device=device,
|
||||
executor=executor,
|
||||
model_fn=cfg.model_fn,
|
||||
**strategy_kwargs,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user