3.1.24.4. unit_scaling.parameter.ParameterData

class unit_scaling.parameter.ParameterData(*args, **kwargs)[source]

Extra fields for torch.nn.Parameter, tagging u-μP metadata.

Objects supporting this protocol should implicitly also support torch.nn.Parameter.