Skip to content

Commit 2312827

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 67f280b commit 2312827

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/lightning/pytorch/callbacks/throughput_monitor.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,7 @@ def setup(self, trainer: "Trainer", pl_module: "LightningModule", stage: str) ->
9595

9696
if stage == TrainerFn.FITTING and trainer.enable_validation:
9797
# `fit` includes validation inside
98-
throughput = Throughput(
99-
available_flops=self.available_flops, world_size=trainer.world_size, **self.kwargs
100-
)
98+
throughput = Throughput(available_flops=self.available_flops, world_size=trainer.world_size, **self.kwargs)
10199
self._throughputs[RunningStage.VALIDATING] = throughput
102100

103101
throughput = Throughput(available_flops=self.available_flops, world_size=trainer.world_size, **self.kwargs)

0 commit comments

Comments
 (0)