INNER CODE UNIT · Python

arguments_nn_layers

apple/ml-cvnets · cvnets/layers/__init__.py:97

def arguments_nn_layers(parser: argparse.ArgumentParser):
    # Retrieve layer specific arguments
    parser = layer_specific_args(parser)

    # activation and normalization arguments
    from cvnets.layers.activation import arguments_activation_fn

    parser = arguments_activation_fn(parser)

    from cvnets.layers.normalization import arguments_norm_layers

    parser = arguments_norm_layers(parser)

    return parser

View source record →