File size: 1,427 Bytes
2f6594f | 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 | {
"name": "david_training",
"run_id": "20251012_181608",
"dataset_name": "AbstractPhil/imagenet-clip-features-orderly",
"model_variant": "clip_vit_laion_bigg14",
"num_classes": 1000,
"preset": "clip_vit_bigg14",
"custom_config_path": null,
"num_classes_override": null,
"use_belly_override": null,
"belly_expand_override": null,
"progressive_training_override": false,
"scale_warmup_epochs_override": null,
"num_epochs": 10,
"batch_size": 1024,
"learning_rate": 0.001,
"weight_decay": 1e-05,
"warmup_epochs": 3,
"use_rose_loss": false,
"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": 10.0,
"scheduler_type": "cosine_restarts",
"min_lr": 1e-06,
"freeze_strategy": "none",
"freeze_threshold": 90.0,
"unfreeze_on_plateau": true,
"patience": 10,
"track_gradients": true,
"gradient_scale_threshold": 1e-05,
"gradient_scale_multiplier": 10.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
} |