Advertisement
Guest User

Untitled

a guest
Aug 24th, 2019
78
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.49 KB | None | 0 0
  1. for key in model.layer2.state_dict():
  2. print('name: ', name)
  3. param = model.layer2.state_dict()[key]
  4. print('type(param): ', type(param))
  5. print('param.shape : ', param.shape)
  6. print('param.requires_grad: ', param.requires_grad)
  7. print('isinstance(param, .nn.Module): ', isinstance(param, nn.Module))
  8. print('isinstance(param, nn.Parameter): ', isinstance(param, nn.Parameter))
  9. print('isinstance(param, torch.Tensor): ', isinstance(param, torch.Tensor))
  10. print('=====')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement