🐛 Bug
named_parameters(prefix: str = '', recurse: bool = True) -> Iterator[Tuple[str, torch.Tensor]] has torch.Tensor in return type, but seems that it always narrower type - nn.Parameter (as in torch.nn.Module.parameters signature)
Environment
pytorch 1.7
cc @albanD @mruberry @ezyang @malfet @rgommers @xuzhao9 @gramster
🐛 Bug
named_parameters(prefix: str = '', recurse: bool = True) -> Iterator[Tuple[str, torch.Tensor]]hastorch.Tensorin return type, but seems that it always narrower type -nn.Parameter(as intorch.nn.Module.parameterssignature)Environment
pytorch 1.7
cc @albanD @mruberry @ezyang @malfet @rgommers @xuzhao9 @gramster