File size: 1,564 Bytes
a5f93a0
 
5e37977
a5f93a0
b4eb156
a5f93a0
b4eb156
a5f93a0
 
 
 
b4eb156
 
 
 
 
 
 
 
 
 
 
 
 
8f5c228
a5f93a0
b4eb156
a5f93a0
8f5c228
a5f93a0
 
 
 
 
 
 
 
 
 
 
74ff46f
 
a5f93a0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
{
  "name": "david_training",
  "run_id": "20251012_060013",
  "dataset_name": "AbstractPhil/imagenet-clip-features-orderly",
  "model_variant": "clip_vit_l14",
  "num_classes": 1000,
  "preset": "clip_vit_l14_deep",
  "custom_config_path": null,
  "num_classes_override": null,
  "use_belly_override": null,
  "belly_expand_override": null,
  "progressive_training_override": true,
  "scale_warmup_epochs_override": {
    "256": 0,
    "512": 1,
    "768": 2,
    "1024": 3,
    "1280": 4,
    "1536": 5,
    "1792": 6,
    "2048": 7,
    "2304": 8,
    "2560": 9
  },
  "num_epochs": 10,
  "batch_size": 1024,
  "learning_rate": 0.001,
  "weight_decay": 1e-05,
  "warmup_epochs": 0,
  "use_rose_loss": true,
  "rose_initial_weight": 0.1,
  "rose_max_weight": 0.5,
  "rose_weight_schedule": "adaptive",
  "use_cayley_loss": false,
  "cayley_weight": 0.001,
  "scale_loss_balance": null,
  "use_mixed_precision": false,
  "gradient_clip": 5.0,
  "scheduler_type": "cosine_restarts",
  "min_lr": 1e-06,
  "freeze_strategy": "never",
  "freeze_threshold": 90.0,
  "unfreeze_on_plateau": true,
  "patience": 10,
  "track_gradients": true,
  "gradient_scale_threshold": 1e-07,
  "gradient_scale_multiplier": 5.0,
  "log_interval": 50,
  "val_interval": 1,
  "save_interval": 5,
  "log_fusion_weights": true,
  "log_loss_components": true,
  "save_format": "safetensors",
  "hf_repo": "AbstractPhil/gated-david",
  "upload_to_hub": true,
  "base_dir": "./david_training",
  "num_workers": 10,
  "pin_memory": true,
  "prefetch_factor": 4,
  "persistent_workers": true
}