spfrommer/torchexplorer

Pretty print tooltips that are shown in Vega

Closed this issue · 1 comments

Ideally, named modules should have their names printed, with new lines for different parameters (e.g., convolution stride, etc). Should also have input/output sizes (this would be per invocation, which might be tricky).

Other things to potentially include: the pamaeter count