import argparse

def get_args_parser():
    parser = argparse.ArgumentParser(description='Optimal Transport AutoEncoder training for Amass',
                                     add_help=True,
                                     formatter_class=argparse.ArgumentDefaultsHelpFormatter)

    parser.add_argument('mm_image_tower', action='store_true', default=True, help='if use multimodal image tower')
    parser.add_argument('mm_video_tower', action='store_true', default=True, help='if use multimodal video tower')

    return parser.parse_args()