mighty.utils.common.find_layers¶
- mighty.utils.common.find_layers(model, layer_class)[source]¶
Find all layers of type layer_class in the input model and yield them as a generator.
- Parameters:
- modelnn.Module
A model.
- layer_classtype
A layer class to look for.
- Yields:
- nn.Module
A children layer of instance layer_class found in the input model.