[BugFix] Torch 1.13 compat (#424)

pytorch · Jun 14, 2023 · 4d679e4 · 4d679e4
1 parent 61cc530
commit 4d679e4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tensordict/nn/functional_modules.py b/tensordict/nn/functional_modules.py
@@ -19,7 +19,12 @@
 
 from tensordict.utils import implement_for
 from torch import nn
-from torch.nn.modules.module import _global_parameter_registration_hooks
+
+try:
+    from torch.nn.modules.module import _global_parameter_registration_hooks
+except ModuleNotFoundError:
+    # old torch version, passing
+    pass
 
 
 @implement_for("torch", "2.0", None)