warlockee's picture
Upload folder using huggingface_hub
bd3ae0e verified
Raw
History Blame Contribute Delete
666 Bytes
augmentation:
feature_dropout: 0.2
temporal_crop: true
temporal_crop_end: true
temporal_crop_ratio: 0.5
backbone: dinov3_vitb16
data_curation:
post_crash_margin: 1
remove_post_crash: true
sample_weights:
enabled: true
negative_weight: 1.0
positive_weight: 5.0
eval:
balanced_val: true
seed: 42
val_ratio: 0.15
max_seq_len: 48
model:
dropout: 0.4
hidden_dim: 128
num_heads: 4
num_layers: 2
pool: mean
type: transformer
training:
batch_size: 32
epochs: 80
label_smoothing: 0.1
lr: 0.0001
optimizer: adamw
patience: 20
pos_weight: 1.0
scheduler: cosine
warmup_epochs: 5
weight_decay: 0.08
truncation: end