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