Skip to content

Commit 2350854

Browse files
committed
Merge branch 'debug_con' of github.com:daniellepintz/pytorch-lightning into debug_con
2 parents 870a984 + b4617e8 commit 2350854

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

pytorch_lightning/trainer/trainer.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
from pytorch_lightning.core.datamodule import LightningDataModule
3434
from pytorch_lightning.core.optimizer import LightningOptimizer
3535
from pytorch_lightning.loggers import LightningLoggerBase
36-
from pytorch_lightning.loggers.base import LoggerCollection, DummyLogger
36+
from pytorch_lightning.loggers.base import DummyLogger, LoggerCollection
3737
from pytorch_lightning.loggers.tensorboard import TensorBoardLogger
3838
from pytorch_lightning.loops import PredictionLoop, TrainingBatchLoop, TrainingEpochLoop
3939
from 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+
21652166
def _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+
)

0 commit comments

Comments
 (0)