add lines between code block

Lightning-AI · Jul 16, 2020 · 6249630 · 6249630
1 parent c7ff87b
commit 6249630
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1728,10 +1728,12 @@ def to_onnx(self, file_path: str, input_sample: Optional[Tensor] = None, **kwarg
 
         Example:
             .. code-block:: python
+
                 file_path = './model.onnx'
                 model = MyLightningModule(...)
                 input_sample = torch.randn((1, 28 * 28))
                 model.to_onnx(file_path, input_sample, export_params=True)
+
         """
 
         if isinstance(input_sample, Tensor):