model_config.json

{
    "backbone_arch": "dinov2_vitb14",
    "backbone_config": {
        "num_trainable_blocks": 4,
        "return_token": true,
        "norm_layer": true
    },
    "agg_arch": "mixvpr",
    "agg_config": {
	        "in_channels": 768,
		    "in_h": 16,
		        "in_w": 16,
			    "out_channels": 1024,
			        "out_rows": 4,
				    "mix_depth": 1
    },
    "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": "MultiSimilarityLoss",
    "miner_name": null,
    "miner_margin": 0.1,
    "faiss_gpu": false
}