v1.5.4

LayerInfo's trainable field is now trainable_params

v1.5.3

• Display layers that share the same variable
  • e.g. activation layers that are defined once and then reused throughout the model
• README updates.

v1.5.2

• Use sys.getsizeof for calculating input size. In the future, we will use this instead of the tensor shape to calculate the size.
• Rework the input_data correction to allow nested dicts and other data structure combinations.
• Add missing basic summary test.
• Refactor the main summary function to use a common traversal helper function.

v1.5.1

• Fix bug causing inconsistent Mult-Add totals that do not sum correctly.
• Overhaul output testing to work automatically for all tests
• Add cache_forward_pass to make it easier to iterate on depths in Jupyter Notebooks

v1.5.0

Upgrade the version number past v1.4.5 in order for pip to resolve the version correctly across older versions of pip.

v0.1.5

• Fix issues with torch.jit scripted modules
• Add support for ParameterLists
• Display bias layers in verbose=2 mode

v0.1.4

Add py.typed to surface type annotations.

v0.1.3

Fix bug with inconsistent calculations for total_params using different depths.

v0.1.2

Fix MACs calculation and differences in output when using different depths.

depth parameter now only affects formatting, calculated values do not change.

v0.1.1

Add model name to the topmost row of the summary table, fixed bug in nested_list_size.