self.forward to self

Lightning-AI · Jul 28, 2020 · eb9b6e4 · eb9b6e4
1 parent a675cb8
commit eb9b6e4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1740,7 +1740,8 @@ def to_onnx(self, file_path: str, input_sample: Optional[Tensor] = None, **kwarg
 
         if 'example_outputs' not in kwargs:
             self.eval()
-            kwargs['example_outputs'] = self.forward(input_data)
+            kwargs['example_outputs'] = self(input_data)
+
         torch.onnx.export(self, input_data, file_path, **kwargs)
 
     @property