We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7862ca4 commit 4770540Copy full SHA for 4770540
docs/source-fabric/guide/compile.rst
@@ -86,7 +86,7 @@ You should always *exclude* the first call to ``forward()`` from your measuremen
86
87
fabric = L.Fabric(accelerator="cuda", devices=1)
88
89
- model = models.densenet121() #.to(torch.float32)
+ model = models.densenet121() # .to(torch.float32)
90
input = torch.randn(16, 3, 128, 128, device=fabric.device)
91
92
compiled_model = torch.compile(model, mode="reduce-overhead")
0 commit comments