understanding_pytorch___20___summary_state_dict.py

for key in model.layer2.state_dict():
    print('name: ', name)
    param = model.layer2.state_dict()[key]
    print('type(param): ', type(param))
    print('param.shape : ', param.shape)
    print('param.requires_grad: ', param.requires_grad)
    print('isinstance(param, .nn.Module): ', isinstance(param, nn.Module))
    print('isinstance(param, nn.Parameter): ', isinstance(param, nn.Parameter))
    print('isinstance(param, torch.Tensor): ', isinstance(param, torch.Tensor))
    print('=====')