| import argparse | |
| parser = argparse.ArgumentParser(description='S3BIR') | |
| parser.add_argument('--exp_name', type=str, default='DINOv2_prompt') | |
| # ---------------------- | |
| # Training Params | |
| # ---------------------- | |
| parser.add_argument('--dinov2_lr', type=float, default=1e-4) | |
| parser.add_argument('--dinov2_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) | |
| # ---------------------- | |
| # 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() |