normalization.py 🚧¢

Source: LION/models/diffusion/NCSNpp_helpers/normalization.py

Warning

This file has not yet received a complete narrative and docstring audit. Its public source-level API is listed automatically below.

Normalization layers.

get_normalization(config, conditional=False)

Obtain normalization modules from the config file.

class ConditionalBatchNorm2d(num_features, num_classes, bias=True)

No docstring is available.

ConditionalBatchNorm2d.forward(self, x, y)

No docstring is available.

class ConditionalInstanceNorm2d(num_features, num_classes, bias=True)

No docstring is available.

ConditionalInstanceNorm2d.forward(self, x, y)

No docstring is available.

class ConditionalVarianceNorm2d(num_features, num_classes, bias=False)

No docstring is available.

ConditionalVarianceNorm2d.forward(self, x, y)

No docstring is available.

class VarianceNorm2d(num_features, bias=False)

No docstring is available.

VarianceNorm2d.forward(self, x)

No docstring is available.

class ConditionalNoneNorm2d(num_features, num_classes, bias=True)

No docstring is available.

ConditionalNoneNorm2d.forward(self, x, y)

No docstring is available.

class NoneNorm2d(num_features, bias=True)

No docstring is available.

NoneNorm2d.forward(self, x)

No docstring is available.

class InstanceNorm2dPlus(num_features, bias=True)

No docstring is available.

InstanceNorm2dPlus.forward(self, x)

No docstring is available.

class ConditionalInstanceNorm2dPlus(num_features, num_classes, bias=True)

No docstring is available.

ConditionalInstanceNorm2dPlus.forward(self, x, y)

No docstring is available.