fastreid.utils.weight_init Namespace Reference

Functions
	weights_init_kaiming (m)
	weights_init_classifier (m)

Detailed Description

@author:  xingyu liao
@contact: sherlockliao01@gmail.com

Function Documentation

weights_init_classifier()

fastreid.utils.weight_init.weights_init_classifier

(

m

)

Definition at line 32 of file weight_init.py.

def weights_init_classifier(m):
    classname = m.__class__.__name__
    if classname.find('Linear') != -1:
        nn.init.normal_(m.weight, std=0.001)
        if m.bias is not None:
            nn.init.constant_(m.bias, 0.0)

weights_init_kaiming()

fastreid.utils.weight_init.weights_init_kaiming

(

m

)

Definition at line 16 of file weight_init.py.

def weights_init_kaiming(m):
    classname = m.__class__.__name__
    if classname.find('Linear') != -1:
        nn.init.normal_(m.weight, 0, 0.01)
        if m.bias is not None:
            nn.init.constant_(m.bias, 0.0)
    elif classname.find('Conv') != -1:
        nn.init.kaiming_normal_(m.weight, mode='fan_out', nonlinearity='relu')
        if m.bias is not None:
            nn.init.constant_(m.bias, 0.0)
    elif classname.find('BatchNorm') != -1:
        if m.affine:
            nn.init.normal_(m.weight, 1.0, 0.02)
            nn.init.constant_(m.bias, 0.0)