File tree Expand file tree Collapse file tree 1 file changed +9
-8
lines changed
pytorch_lightning/trainer Expand file tree Collapse file tree 1 file changed +9
-8
lines changed Original file line number Diff line number Diff line change 3333from pytorch_lightning .core .datamodule import LightningDataModule
3434from pytorch_lightning .core .optimizer import LightningOptimizer
3535from pytorch_lightning .loggers import LightningLoggerBase
36- from pytorch_lightning .loggers .base import LoggerCollection , DummyLogger
36+ from pytorch_lightning .loggers .base import DummyLogger , LoggerCollection
3737from pytorch_lightning .loggers .tensorboard import TensorBoardLogger
3838from pytorch_lightning .loops import PredictionLoop , TrainingBatchLoop , TrainingEpochLoop
3939from pytorch_lightning .loops .dataloader .evaluation_loop import EvaluationLoop
@@ -2165,11 +2165,12 @@ def terminate_on_nan(self, val: bool) -> None:
21652165 )
21662166 self ._terminate_on_nan = val # : 212
21672167
2168+
21682169def _determine_batch_limits (batches : Union [int , float ], name : str ) -> Union [int , float ]:
2169- if 0 <= batches <= 1 :
2170- return batches
2171- if batches > 1 and batches % 1.0 == 0 :
2172- return int (batches )
2173- raise MisconfigurationException (
2174- f"You have passed invalid value { batches } for { name } , it has to be in [0.0, 1.0] or an int."
2175- )
2170+ if 0 <= batches <= 1 :
2171+ return batches
2172+ if batches > 1 and batches % 1.0 == 0 :
2173+ return int (batches )
2174+ raise MisconfigurationException (
2175+ f"You have passed invalid value { batches } for { name } , it has to be in [0.0, 1.0] or an int."
2176+ )
You can’t perform that action at this time.
0 commit comments