Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update src/lightning/fabric/strategies/strategy.py

This commit is contained in:
Luca Antiga 2024-11-25 11:35:10 +01:00 committed by GitHub
parent 2d347d0bbb
commit 5d227ff480
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/lightning/fabric/strategies/strategy.py
View File

@ -147,7 +147,7 @@ class Strategy(ABC):
def setup_module_and_optimizers(
self, module: Module, optimizers: list[Optimizer], scheduler: Optional[_LRScheduler] = None
) -> Tuple[Module, list[Optimizer], Optional[_LRScheduler]]:
) -> tuple[Module, list[Optimizer], Optional[_LRScheduler]]:
"""Set up a model and multiple optimizers together.
The returned objects are expected to be in the same order they were passed in. The default implementation will