bob.med.tb.utils.summary¶
Functions
|
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