File size: 675 Bytes
3e0e9f4 09892bf 3e0e9f4 09892bf 3e0e9f4 09892bf 3e0e9f4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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() |