import argparse parser = argparse.ArgumentParser(description='Sketch-based OD') parser.add_argument('--exp_name', type=str, default='LN_prompt') # ---------------------- # Training Params # ---------------------- parser.add_argument('--clip_lr', type=float, default=1e-4) parser.add_argument('--clip_LN_lr', type=float, default=1e-6) parser.add_argument('--prompt_lr', type=float, default=1e-4) parser.add_argument('--linear_lr', type=float, default=1e-4) parser.add_argument('--model_type', type=str, default='one_encoder', choices=['one_encoder', 'two_encoder']) # ---------------------- # ViT Prompt Parameters # ---------------------- parser.add_argument('--prompt_dim', type=int, default=768) parser.add_argument('--n_prompts', type=int, default=3) opts = parser.parse_args()