INNER CODE UNIT · Python

modeling_arguments

apple/ml-cvnets · cvnets/__init__.py:21

def modeling_arguments(parser: argparse.ArgumentParser) -> argparse.ArgumentParser:
    # text encoder arguments (usually for multi-modal tasks)
    parser = arguments_text_encoder(parser)
    # image projection head arguments (usually for multi-modal tasks)
    parser = arguments_image_projection_head(parser)
    # model arguments
    parser = arguments_model(parser)
    # neural network layer argumetns
    parser = arguments_nn_layers(parser)
    # EMA arguments
    parser = arguments_ema(parser)
    # anchor generator arguments (for object detection)
    parser = arguments_anchor_gen(parser)
    # box matcher arguments (for object detection)
    parser = arguments_box_matcher(parser)
    # neural aug arguments
    parser = arguments_neural_augmentor(parser)

View source record →