Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add size information #21

Merged
merged 1 commit into from
Aug 9, 2018
Merged

Add size information #21

merged 1 commit into from
Aug 9, 2018
+18 −5

Commits on Aug 4, 2018

  1. Add size information 

    Added information for estimating the total size of the model. Estimates taken from here: http://jacobkimmel.github.io/pytorch_estimating_model_size/
-calculates size of input, parameters, and forward/backward pass intermediate variables
-prints out these estimates and total
-batch_size optional input
    @rmchurch
    rmchurch authored Aug 4, 2018
    Configuration menu
    Copy the full SHA
    81df232 View commit details
    Browse the repository at this point in the history