Inheritance diagram for fastreid.layers.batch_norm.BatchNorm:

Public Member Functions
	__init__ (self, num_features, eps=1e-05, momentum=0.1, weight_freeze=False, bias_freeze=False, weight_init=1.0, bias_init=0.0, **kwargs)

Public Attributes
	weight

	bias

Detailed Description

Definition at line 23 of file batch_norm.py.

Constructor & Destructor Documentation

◆ init()

fastreid.layers.batch_norm.BatchNorm.__init__	(		self,
			num_features,
			eps = 1e-05,
			momentum = 0.1,
			weight_freeze = False,
			bias_freeze = False,
			weight_init = 1.0,
			bias_init = 0.0,
		**	kwargs )

Reimplemented in fastreid.layers.batch_norm.FrozenBatchNorm, and fastreid.layers.batch_norm.GhostBatchNorm.

Definition at line 24 of file batch_norm.py.

                 bias_init=0.0, **kwargs):
        super().__init__(num_features, eps=eps, momentum=momentum)
        if weight_init is not None: nn.init.constant_(self.weight, weight_init)
        if bias_init is not None: nn.init.constant_(self.bias, bias_init)
        self.weight.requires_grad_(not weight_freeze)
        self.bias.requires_grad_(not bias_freeze)
 
 

Member Data Documentation

◆ bias

fastreid.layers.batch_norm.BatchNorm.bias

Definition at line 28 of file batch_norm.py.

◆ weight

fastreid.layers.batch_norm.BatchNorm.weight

Definition at line 27 of file batch_norm.py.

The documentation for this class was generated from the following file:

smreid/fastreid/layers/batch_norm.py

Public Member Functions