| { |
| "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 |
| } |
|
|
