fastreid.layers.context_block Namespace Reference

Classes
class	ContextBlock

Functions
	last_zero_init (m)

Function Documentation

last_zero_init()

fastreid.layers.context_block.last_zero_init

(

m

)

Definition at line 9 of file context_block.py.

def last_zero_init(m):
    if isinstance(m, nn.Sequential):
        nn.init.constant_(m[-1].weight, val=0)
        if hasattr(m[-1], 'bias') and m[-1].bias is not None:
            nn.init.constant_(m[-1].bias, 0)
    else:
        nn.init.constant_(m.weight, val=0)
        if hasattr(m, 'bias') and m.bias is not None:
            nn.init.constant_(m.bias, 0)