Add au-posw-cw checkpoint
Browse files- au-posw-cw/config.json +32 -0
- au-posw-cw/model.pt +3 -0
au-posw-cw/config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"train_annotations": "data/training_set_annotations.txt",
|
| 3 |
+
"val_annotations": "data/validation_set_annotations.txt",
|
| 4 |
+
"image_root": "data/cropped_aligned",
|
| 5 |
+
"output_dir": "/data3/salah/projects/ABAW2026/code/runs/ablation_e20_affectflow_dinov3_clsw_auposw_e20",
|
| 6 |
+
"backbone": "dinov3_vits16",
|
| 7 |
+
"backbone_weights": null,
|
| 8 |
+
"strict_backbone": true,
|
| 9 |
+
"image_size": 224,
|
| 10 |
+
"batch_size": 64,
|
| 11 |
+
"epochs": 20,
|
| 12 |
+
"lr": 0.0001,
|
| 13 |
+
"weight_decay": 0.0001,
|
| 14 |
+
"num_workers": 4,
|
| 15 |
+
"freeze_backbone": true,
|
| 16 |
+
"deterministic_weight": 1.0,
|
| 17 |
+
"flow_weight": 1.0,
|
| 18 |
+
"va_weight": 1.0,
|
| 19 |
+
"expr_weight": 1.0,
|
| 20 |
+
"au_weight": 1.0,
|
| 21 |
+
"focal_gamma": 0.0,
|
| 22 |
+
"label_smoothing": 0.0,
|
| 23 |
+
"separate_proj_heads": false,
|
| 24 |
+
"expr_class_weighted": true,
|
| 25 |
+
"balanced_sampler": false,
|
| 26 |
+
"au_pos_weighted": true,
|
| 27 |
+
"flow_steps": 20,
|
| 28 |
+
"flow_samples": 8,
|
| 29 |
+
"seed": 42,
|
| 30 |
+
"device": "cuda",
|
| 31 |
+
"smoke_test": false
|
| 32 |
+
}
|
au-posw-cw/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0bda4806ab206c3b3b3b8c00b9ce0384d74a6977c00f8c4a0fb860a25224ea6f
|
| 3 |
+
size 148623166
|