-
Notifications
You must be signed in to change notification settings - Fork 25.6k
Open
Labels
module: nnRelated to torch.nnRelated to torch.nntriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
The Parametrization that contains the original Tensor is not traversing itself properly during the forward.
In particular here:
pytorch/torch/nn/utils/parametrize.py
Line 265 in 96cb05b
for module in list(self._modules.values())[1:]: |
We use all the _modules
even the ones that are not indexed by integers.
This prevent parametrization of parametrization because it would add a parametrizations
submodule to this and the forward would try to traverse it.
Metadata
Metadata
Assignees
Labels
module: nnRelated to torch.nnRelated to torch.nntriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module