From e0bcfee556ed6e430011f0b8f8dd1e8f455c5532 Mon Sep 17 00:00:00 2001 From: ZXConjure <2632590648@qq.com> Date: Thu, 22 Jan 2026 17:25:39 +0800 Subject: [PATCH] Fix: add **kwargs to _named_members for PyTorch 2.6+ compatibility PyTorch 2.6+ introduces a 'remove_duplicate' argument in named_parameters. Adding **kwargs allows _named_members to accept future arguments without crashing. Verified locally on PyTorch 2.7. --- tensorlayerx/nn/core/core_torch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorlayerx/nn/core/core_torch.py b/tensorlayerx/nn/core/core_torch.py index 97cedb5..9f44f5a 100644 --- a/tensorlayerx/nn/core/core_torch.py +++ b/tensorlayerx/nn/core/core_torch.py @@ -125,7 +125,7 @@ def _call_impl_tlx(self, *input, **kwargs): # TODO RNN enabled after repair __call__: Callable[..., Any] = _call_impl_tlx - def _named_members(self, get_members_fn, prefix='', recurse=True): + def _named_members(self, get_members_fn, prefix='', recurse=True, **kwargs): r"""Helper method for yielding various names + members of modules.""" memo = set() modules = self.named_modules(prefix=prefix) if recurse else [(prefix, self)]