mighty.utils.hooks.get_layers_ordered

mighty.utils.hooks.get_layers_ordered(model, input_sample, ignore_layers=(<class 'torch.nn.modules.container.Sequential'>, ), ignore_children=())

Returns a list of ordered layers of the input model.

Parameters:

modelnn.Module

An input model.

input_sampletorch.Tensor

A sample tensor to be used with the model.

ignore_layerstuple of type

A tuple of model classes not to include in the final result. Default: (nn.Sequential,)

ignore_childrentuple of type

A tuple of model classes to skip entering their children. Default: ()

Returns:

layers_orderedlist of torch.Tensor

An ordered list of layers of the input model. Note that some modules might be added in the list more than once.