mednet.utils.summary

Functions

summary(model)

Count the number of parameters in each model layer.

mednet.utils.summary.summary(model)[source]

Count the number of parameters in each model layer.

Parameters:

model (Module) – Model to summarize.

Returns:

A tuple containing a multiline string representation of the network and the number of parameters.

Return type:

tuple[int, str]