deepdraw.utils.summary#

Functions

summary(model)

Counts the number of parameters in each model layer.
deepdraw.utils.summary.summary(model)[source]#

Counts the number of parameters in each model layer.

Parameters:

model (torch.nn.Module) – model to summarize

Returns:

  • repr (str) – a multiline string representation of the network

  • nparam (int) – number of parameters