hookeai.material_model_finder.utilities.torch_parameters.get_model_parameter_dict

get_model_parameter_dict(model)

Store torch Module parameters in torch.nn.ParameterDict.

Nested modules or submodules are stored as nested dictionaries.

Parameters:

model (torch.nn.Module) – PyTorch model.

Returns:

param_dict – Parameter dictionary.

Return type:

torch.nn.ParameterDict()