UniformInit

class mmengine.model.UniformInit(a=0, b=1, **kwargs)

Initialize module parameters with values drawn from the uniform distribution \(\mathcal{U}(a, b)\).

Parameters

• a (int | float) – the lower bound of the uniform distribution. Defaults to 0.

• b (int | float) – the upper bound of the uniform distribution. Defaults to 1.

• bias (int | float) – the value to fill the bias. Defaults to 0.

• bias_prob (float, optional) – the probability for bias initialization. Defaults to None.

• layer (str | list[str], optional) – the layer will be initialized. Defaults to None.