Function create_act_layer

timm/models/layers/create_act.py:149–153

(name: Union[nn.Module, str], inplace=None, **kwargs)

def create_act_layer(name: Union[nn.Module, str], inplace=None, **kwargs):
    act_layer = get_act_layer(name)
    if act_layer is None:
        return None
    return act_layer(**kwargs) if inplace is None else act_layer(inplace=inplace, **kwargs)