Skip to content

nexport 0.3.0 (The Parameter Update)
Compare
Choose a tag to compare
@JordanWelsman JordanWelsman released this 19 Jan 04:34
· 85 commits to main since this release
v0.3.0
d9ec643
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Description

This update brings useful functions for calculating a PyTorch model's number of trainable parameters, layers and neurons. The user has an option to filter the results based on whether they want to include those in the input & output layers.

Changelog

  • New functions
    • calculate_layers()
    • calculate_neurons()
    • calculate_parameters()
  • Cleaner directory structure
    • Example PyTorch networks moved to separate models.py file
Assets 3
Loading