layer_finder¶

probly.traverse_nn.utils.layer_finder(obj: object, state: State[Any]) → TraverserResult[object][source]¶: Collect visited objects that are instances of LAYER_TYPES into FOUND_LAYERS.