bob.med.tb.utils.summary

Functions

summary(model)

Counts the number of parameters in each model layer

bob.med.tb.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