Shortcuts

UniformInit

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

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.

Read the Docs v: v0.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.