Skip to content

Instantly share code, notes, and snippets.

@bilal2vec
Created February 27, 2020 20:31
Show Gist options
  • Save bilal2vec/522f4c5d99cecf7ad740e1302fc946c0 to your computer and use it in GitHub Desktop.
Save bilal2vec/522f4c5d99cecf7ad740e1302fc946c0 to your computer and use it in GitHub Desktop.
# Print out the number of parameters in a Pytorch model
print(n_params(model))
# 150909673
# Save a model for a particular cross-validation fold to disk
save_model(model, fold=0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment