amend

pytorch · Nov 21, 2023 · ab88669 · ab88669
1 parent 3e976ac
commit ab88669
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensordict/nn/common.py b/tensordict/nn/common.py
@@ -1091,7 +1091,7 @@ def __init__(
             )
 
         self.module = module
-        # make_functional(self, keep_params=True, return_params=False)
+        make_functional(self, keep_params=True, return_params=False)
 
     @property
     def is_functional(self) -> bool: