search

jacobkimmel / pytorch_modelsize

Estimates the size of a PyTorch model in memory
MIT License
354 stars 49 forks source link

fix bits parameter in init function #6

Closed phillies closed 4 years ago

phillies commented 4 years ago

the bits parameter in the init function was not stored as class member

jacobkimmel commented 4 years ago

Ah, isn't that a foolish bug!

Thanks for the fix!