hookeai.material_model_finder.utilities.torch_parameters¶

Utilities to handle Torch parameters.

Functions¶

set_parameter_in_dict: Store parameter in torch.nn.ParameterDict.
get_model_parameter_dict: Store torch Module parameters in torch.nn.ParameterDict.

Functions

`get_model_parameter_dict`(model)	Store torch Module parameters in torch.nn.ParameterDict.
`set_parameter_in_dict`(param_dict, ...)	Store parameter in torch.nn.ParameterDict.