KilianFt commited on
Commit
471892d
·
verified ·
1 Parent(s): 624e574

Upload FlashJudge artifacts from run_20260428_111358

Browse files
flash_judge_checkpoint.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ce7ad46d9f629a2862024d84efbf474beb5ecee91bef6e992bf82092d3b545f
3
+ size 1534188467
training_config.json ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "google/siglip2-base-patch16-naflex",
3
+ "dataset": "apartments",
4
+ "apartments_jsonl": "data/apartments/question-image-dataset_labeled_expanded.jsonl",
5
+ "apartments_images_dir": "data/apartments/images",
6
+ "fmnist_root": "data",
7
+ "output_root": "models",
8
+ "run_name": null,
9
+ "batch_size": 20,
10
+ "gradient_accumulation_steps": 1,
11
+ "stack_size": 1,
12
+ "epochs": 5,
13
+ "max_steps": null,
14
+ "eval_interval": 1,
15
+ "eval_batches": 2,
16
+ "checkpoint_interval": 1000,
17
+ "checkpoint_mode": "max_val_acc",
18
+ "val_ratio": 0.1,
19
+ "seed": 42,
20
+ "apartments_train_group_limit": null,
21
+ "apartments_train_question_limit": null,
22
+ "apartments_question_split_mode": "disjoint",
23
+ "apartments_hard_negative_strategy": "within_group_overlap",
24
+ "apartments_hard_negative_topk": 0,
25
+ "apartments_hard_negative_oversample_factor": 0,
26
+ "num_workers_train": 4,
27
+ "num_workers_val": 2,
28
+ "text_max_length": 64,
29
+ "image_max_num_patches": null,
30
+ "lr_backbone": 1e-6,
31
+ "lr_head": 0.0002,
32
+ "weight_decay": 0.01,
33
+ "freeze_backbone": false,
34
+ "freeze_backbone_epochs": 5,
35
+ "freeze_backbone_steps": null,
36
+ "backbone_trainable_scope": "full",
37
+ "attention_heads": 2,
38
+ "attention_dropout": 0.3,
39
+ "image_self_attention_layers": 0,
40
+ "aggregation_mode": "attention",
41
+ "head_feature_mode": "matching",
42
+ "late_interaction_topk": 8,
43
+ "head_hidden_dim": 768,
44
+ "head_num_layers": 2,
45
+ "head_dropout": 0.2,
46
+ "head_activation": "swiglu",
47
+ "label_smoothing": 0.05,
48
+ "alignment_loss_weight": 0.05,
49
+ "alignment_loss_logit_scale": 18.0,
50
+ "alignment_loss_hard_negative_topk": 1,
51
+ "compile_model": false,
52
+ "compile_fallback_to_eager": true,
53
+ "backbone_torch_dtype": "auto",
54
+ "attn_implementation": null,
55
+ "resume_path": null,
56
+ "min_free_space_gb": 1.0,
57
+ "use_wandb": true,
58
+ "wandb_project": "flash-judge",
59
+ "wandb_entity": null,
60
+ "wandb_run_name": null,
61
+ "save_artifacts": true,
62
+ "push_to_hub": true,
63
+ "hub_org": "FuncAI",
64
+ "hub_repo_name": "FlashJudgev5_frozen",
65
+ "hub_private": false,
66
+ "hub_token": null,
67
+ "device": null,
68
+ "use_amp": true
69
+ }
training_logs.csv ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ epoch,train_loss,train_acc,val_loss,val_acc,max_val_acc
2
+ 1,0.5482786992188536,0.7919552711904203,0.44720758710316094,0.7876536956392839,0.7876536956392839
3
+ 2,0.4527545233316229,0.8569891210769351,0.49181961134022684,0.7818391476681487,0.7876536956392839
4
+ 3,0.3874157649465124,0.8966062064647413,0.5176910986688721,0.7817879174759642,0.7876536956392839
5
+ 4,0.3331221164834465,0.9286364169992956,0.5424434945223945,0.7795594336739817,0.7876536956392839
6
+ 5,0.29440113875035007,0.9511133286373953,0.581797330123067,0.7810194752499705,0.7876536956392839