Helper Module for Deep Learning.

Module that provides tools to display a graph.

pynet.plotting.network.plot_net(model, shape, static=True, outfileroot=None)[source]¶

Save a PDF file containing the network graph representation.

Sometimes the ‘get_trace_graph’ pytorch function fails: use the ‘plot_net_rescue’ function insteed.

Parameters

model: Net

the network model.

shape: list of int

the shape of a classical input batch dataset.

static: bool, default True

create a static or dynamic view.

outfileroot: str, default None

the file path without extension to generate PDF.

Returns

outfile: str

the path to the generated PDF.

pynet.plotting.network.plot_net_rescue(model, shape, outfileroot=None)[source]¶

Save a PNG file containing the network graph representation.