Model.named_parameters() will lose some layer modules

5

Thanks. I understand what you mean, but I create all the layers by using class LayerName(nn.Module) and there is no any other parameters beyond the torch.nn.layers.
I think they are registered properly and can be visible.

I change the method to group the parameters via model.state_dict() like this:

for k, v in model.state_dict().items():
    name = k
    param = torch.nn.Parameter(v)

Does this way can be seemed as same with named_parameters()?