Skip to content

Commit 78b7a39

Browse files
authored
Update throughput docs (#19415)
1 parent 28a8023 commit 78b7a39

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

docs/source-fabric/index.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,7 @@ Get Started
211211
Loggers <api/loggers>
212212
Precision <api/precision>
213213
Strategies <api/strategies>
214+
Utilities <api/utilities>
214215

215216

216217
.. toctree::

src/lightning/pytorch/callbacks/throughput_monitor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def setup(self, stage):
5050
model = MyModel()
5151
5252
def sample_forward():
53-
batch = torch.randn(...)
53+
batch = torch.randn(..., device="meta")
5454
return model(batch)
5555
5656
self.flops_per_batch = measure_flops(model, sample_forward, loss_fn=torch.Tensor.sum)

0 commit comments

Comments
 (0)