Skip to content

Releases: LucaBonfiglioli/nnviz

1.0.0

17 Jun 11:47
7d9f566
Compare
Choose a tag to compare

Added:

Changed:

  • General API cleanup and testing.

0.4.0

12 Mar 10:51
fa6516e
Compare
Choose a tag to compare

Added:

  • Partial tracing. You can decide which parts of the model to plot individually, without plotting the entire model to avoid huge clusterfucks.
  • Better error handling. Some errors were simply incomprehensible by the average user to the point they become hostile.
  • [dev] DataSpec visitors now have an option to retain control over the visiting process. By default they delegate to the visited.
  • Graphs now have an header with model title and some useful info.
  • Graphs can be now be exported to / loaded from json files.
  • Better collapsing (add more options than just max-depth)
  • Drawer customization (to some extent)

0.3.1

26 Feb 09:54
86ba937
Compare
Choose a tag to compare

Bugfixes:

  • The readme file did not include graphviz-dev as a requirement.
  • The typer version constraint was too loose, due to the "pretty_exception_enable" option introduced later than 0.4.0. Now it's constrained to >=0.6.0.

0.3.0

25 Feb 20:24
65d303c
Compare
Choose a tag to compare

Added:

  • The nodes now also display the number of trainable parameters (along with the percentage relative to the total)

Bugfixes:

  • If a node is used more than once in the same graph, it is now handled properly by nnviz.
  • I/O spec inspection now also works on "call_function", "call_method" and "get_attr" nodes.
  • If an edge gets collapsed, its spec will be handled properly.
  • Added extra width to tables to avoid overlaps with other edges or cluster walls
  • Graphviz init_rank shit is no more an issue. label -> xlabel seemed to do the trick. 🤡