Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Select an option

  • Save n0obcoder/ddd396a725663b990b9ddbefa100b440 to your computer and use it in GitHub Desktop.

Select an option

Save n0obcoder/ddd396a725663b990b9ddbefa100b440 to your computer and use it in GitHub Desktop.
for name, parameter in model.named_parameters():
print('name : {}'.format(name))
print('parameter : {}'.format(parameter.item()))
print('learnable : {}'.format(parameter.requires_grad))
print('parameter.shape: {}'.format(parameter.shape))
print('---------------------------------')
name : linear.weight
parameter : -0.19329047203063965
learnable : True
parameter.shape: torch.Size([1, 1])
---------------------------------
name : linear.bias
parameter : -0.36674630641937256
learnable : True
parameter.shape: torch.Size([1])
---------------------------------
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment