15laddoo's picture
Add DaViT-Base four-head checkpoint (seed 42), safetensors, training meta and card
b5d1055 verified
Raw History Blame Contribute Delete
1.17 kB
{
"backbone": "davit_base.msft_in1k",
"config_file": "configs/davit_base.yaml",
"seed": 42,
"git_commit": "d25b21557f37e206c7fbc7a8612275bc773eb04d-dirty",
"manifest_rows": 108486,
"manifest_sha256": "ecff23f0686067fde9781fdbdd6f5361d699aa6110e93099a7752ed77338c731",
"co_occur_floor": 5,
"world_size": 4,
"per_device_batch_size": 128,
"grad_accum": 1,
"effective_batch_size": 512,
"epochs": 25,
"total_optimizer_steps": 4050,
"warmup_steps": 121,
"learning_rate": 0.0003,
"weight_decay": 0.05,
"label_smoothing": 0.1,
"loss_weights": {
"crop": 1.5,
"category": 0.5,
"disease": 1.0,
"pest": 1.2
},
"precision": "bf16",
"image_size": 224,
"topk_crop_mask": 1,
"vocab_sizes": {
"crop": 110,
"category": 3,
"disease": 285,
"pest": 92
},
"train_data": {
"split_rows": 82991,
"dropped_no_url": 16,
"dropped_image_absent": 0,
"kept": 82975,
"from_scaled_cache": 82975
},
"val_data": {
"split_rows": 9220,
"dropped_no_url": 6,
"dropped_image_absent": 0,
"kept": 9214,
"from_scaled_cache": 9214
},
"train_disease_masked_out": 317,
"val_disease_masked_out": 45,
"torch": "2.8.0+cu128",
"gpu": "NVIDIA L40S"
}