Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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('=====')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement