def build_relu()

in mobile_cv/arch/fbnet_v2/basic_blocks.py [0:0]

• 21 lines of code
• 11 McCabe index (conditional complexity)

def build_relu(name=None, num_channels=None, **kwargs):
    inplace = kwargs.pop("inplace", True)
    if name is None or name == "none":
        return None
    if name == "relu":
        return nn.ReLU(inplace=inplace)
    if name == "relu6":
        return nn.ReLU6(inplace=inplace)
    if name == "leakyrelu":
        return nn.LeakyReLU(inplace=inplace, **kwargs)
    if name == "prelu":
        return nn.PReLU(num_parameters=num_channels, **kwargs)
    if name == "hswish":
        return HSwish()
    if name == "swish":
        return Swish()
    if name in ["sig", "sigmoid"]:
        return nn.Sigmoid()
    if name in ["hsig", "hsigmoid"]:
        return HSigmoid()

    return RELU_REGISTRY.get(name)(**kwargs)