Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- for name, child in model.named_children():
- print('name: ', name)
- 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('=====')
- # Also try model.children(). It doesn't return the name of the children, but the children (nn.Module objects)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement