find_hidden_groups_from_module#
- class braintrace.find_hidden_groups_from_module(model, *model_args, **model_kwargs)[source]#
Find hidden groups from the model.
- Parameters:
model (
Module) – The model.model_args – The model arguments.
model_kwargs – The model keyword arguments.
- Returns:
Sequence of HiddenGroup objects
PrettyDict mapping hidden state paths to hidden groups
- Return type:
Tuple[Sequence[HiddenGroup],PrettyDict]