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
@@ -2162,11 +2162,12 @@ def terminate_on_nan(self, val: bool) -> None:
21622162 )
21632163 self ._terminate_on_nan = val # : 212
21642164
2165+
21652166def _determine_batch_limits (batches : Union [int , float ], name : str ) -> Union [int , float ]:
2166- if 0 <= batches <= 1 :
2167- return batches
2168- if batches > 1 and batches % 1.0 == 0 :
2169- return int (batches )
2170- raise MisconfigurationException (
2171- f"You have passed invalid value { batches } for { name } , it has to be in [0.0, 1.0] or an int."
2172- )
2167+ if 0 <= batches <= 1 :
2168+ return batches
2169+ if batches > 1 and batches % 1.0 == 0 :
2170+ return int (batches )
2171+ raise MisconfigurationException (
2172+ f"You have passed invalid value { batches } for { name } , it has to be in [0.0, 1.0] or an int."
2173+ )
You can’t perform that action at this time.
0 commit comments