register_parameter

UNet.register_parameter(name: str, param: Optional[torch.nn.parameter.Parameter]) → None

Adds a parameter to the module.

The parameter can be accessed as an attribute using given name.

Parameters

• name (string) – name of the parameter. The parameter can be accessed from this module using the given name

• param (Parameter) – parameter to be added to the module.