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()
|