Spaces:
Sleeping
Sleeping
File size: 621 Bytes
6146368 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
seed: -1
dataset: MCAC
test_split: val
MCAC_crop_size: 672
MCAC_occ_limit: 70
MCAC_occ_limit_exemplar: 30
img_channels: 3
image_transforms: ref_rot
counting_backbone: "vit_dino"
counting_backbone_unfreeze_layers: -1
counting_head: "5_32"
upsample_padding_mode: 'replicate' # 'zeros', 'reflect', 'replicate'
matching_type: density # no_match, count, density
matcher_cost_power: 2
normalize_matching: True
counting_loss: pixelwise_mae
gtd_scale: 400
learning_rate: 3e-5
scheduler: StepLR
scheduler_steps: 35
scheduler_gamma: 0.5
weight_decay: 0
train_batch_size: 2
eval_batch_size: 2
val_every: 1
max_epochs: 150 |