Use the setter in the children recursively (#14724)

Author: carmocca

src/pytorch_lightning/CHANGELOG.md

@@ -5,12 +5,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 
-## [1.7.7] - 2022-09-??
+## [1.7.7] - 2022-09-22
 
 ### Fixed
 
 - Fixed the availability check for the neptune-client package ([#14714](https://github.com/Lightning-AI/lightning/pull/14714))
 - Break HPU Graphs into two parts (forward + backward as one and optimizer as another) for better performance ([#14656](https://github.com/Lightning-AI/lightning/pull/14656))
+- Fixed torchscript error with ensembles of LightningModules ([#14657](https://github.com/Lightning-AI/lightning/pull/14657), [#14724](https://github.com/Lightning-AI/lightning/pull/14724))
 
 
 ## [1.7.6] - 2022-09-13
@@ -25,7 +26,6 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 - Fixed an issue to keep downscaling the batch size in case there hasn't been even a single successful optimal batch size with `mode="power"` ([#14372](https://github.com/Lightning-AI/lightning/pull/14372))
 - Fixed an issue where `self.log`-ing a tensor would create a user warning from PyTorch about cloning tensors ([#14599](https://github.com/Lightning-AI/lightning/pull/14599))
 - Fixed compatibility when `torch.distributed` is not available ([#14454](https://github.com/Lightning-AI/lightning/pull/14454))
-- Fixed torchscript error with ensembles of LightningModules ([#14657](https://github.com/Lightning-AI/lightning/pull/14657))
 
 
 ## [1.7.5] - 2022-09-06

src/pytorch_lightning/core/module.py

@@ -297,7 +297,7 @@ def _running_torchscript(self) -> bool:
     def _running_torchscript(self, value: bool) -> None:
         for v in self.children():
             if isinstance(v, LightningModule):
-                v._running_torchscript_internal = value
+                v._running_torchscript = value
         self._running_torchscript_internal = value
 
     def _call_batch_hook(self, hook_name: str, *args: Any) -> Any:

tests/tests_pytorch/models/test_torchscript.py

@@ -174,14 +174,22 @@ def test_torchscript_with_no_input(tmpdir):
 
 
 def test_torchscript_script_recursively():
-    class Child(LightningModule):
+    class GrandChild(LightningModule):
         def __init__(self):
             super().__init__()
             self.model = torch.nn.Linear(1, 1)
 
         def forward(self, inputs):
             return self.model(inputs)
 
+    class Child(LightningModule):
+        def __init__(self):
+            super().__init__()
+            self.model = GrandChild()
+
+        def forward(self, inputs):
+            return self.model(inputs)
+
     class Parent(LightningModule):
         def __init__(self):
             super().__init__()