def weight_reset()

in ml3/learnable_losses.py [0:0]

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

def weight_reset(m):
    if isinstance(m, nn.Conv2d) or isinstance(m, nn.Linear):
        m.reset_parameters()