self.forward to self

Author: lezwon

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