def weight_init()

in ml3/learnable_losses.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

def weight_init(module):
    if isinstance(module, nn.Linear):
        nn.init.xavier_uniform_(module.weight, gain=1.0)
        if module.bias is not None:
            module.bias.data.zero_()