Skip to content

Instantly share code, notes, and snippets.

@alinaselega

alinaselega/print_model_params.py

Created April 6, 2021 17:15
Show Gist options
  • Save alinaselega/b14b37f696a071b69f7840f26e7b6a2e to your computer and use it in GitHub Desktop.
Save alinaselega/b14b37f696a071b69f7840f26e7b6a2e to your computer and use it in GitHub Desktop.
Download ZIP
Code snippet to print model parameters after training
Raw
print_model_params.py
for param_name, param in model.named_parameters():
print(f'Parameter name: {param_name:42} value = {param.item():.4f}')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment