File size: 549 Bytes
445d3d1
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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()