AffectFlow-DINO / finetune /config.json
Bekhouche's picture
Add finetune checkpoint
af7c3eb verified
Raw
History Blame Contribute Delete
686 Bytes
{
"train_annotations": "data/training_set_annotations.txt",
"val_annotations": "data/validation_set_annotations.txt",
"image_root": "data/cropped_aligned",
"output_dir": "code/runs/ablation_e9_affectflow_dinov3_finetune_lr1e-5_e20",
"backbone": "dinov3_vits16",
"backbone_weights": null,
"strict_backbone": true,
"image_size": 224,
"batch_size": 32,
"epochs": 20,
"lr": 1e-05,
"weight_decay": 0.0001,
"num_workers": 4,
"freeze_backbone": false,
"deterministic_weight": 1.0,
"flow_weight": 1.0,
"va_weight": 1.0,
"expr_weight": 1.0,
"au_weight": 1.0,
"flow_steps": 20,
"flow_samples": 8,
"seed": 42,
"device": "cuda",
"smoke_test": false
}