model_config.json

{
    "backbone_arch": "dinov2_vitb14",
    "backbone_config": {
        "num_trainable_blocks": 4,
        "return_token": true,
        "norm_layer": true
    },
    "agg_arch": "salad_multivector",
    "agg_config": {
        "num_channels": 768,
        "num_clusters": 64,
        "cluster_dim": 128
    },
    "lr": 6e-05,
    "optimizer": "adamw",
    "weight_decay": 9.5e-09,
    "momentum": 0.9,
    "lr_sched": "linear",
    "lr_sched_args": {
        "start_factor": 1,
        "end_factor": 0.2,
        "total_iters": 4000
    },
    "loss_name": "MaxSimMSLoss",
    "miner_name": null,
    "miner_margin": 0.1,
    "faiss_gpu": false
}