gated-david / best_model.json
AbstractPhil's picture
Update metrics - Run 20251012_145649
bc514d0 verified
raw
history blame
1.48 kB
{
"model_name": "David-hierarchical-progressive",
"run_id": "20251012_145649",
"timestamp": "2025-10-12T15:11:05.426666",
"best_val_acc": 76.812,
"best_epoch": 7,
"final_train_acc": 88.98340341266986,
"final_train_loss": 1.5605441822221104,
"scale_accuracies": {
"256": 70.048,
"512": 74.262,
"768": 75.674,
"1024": 75.564
},
"architecture": {
"preset": "hierarchical_refinement",
"sharing_mode": "hierarchical",
"fusion_mode": "progressive",
"scales": [
256,
512,
768,
1024
],
"feature_dim": 512,
"num_classes": 1000,
"use_belly": true,
"belly_expand": 2.0
},
"training": {
"dataset": "AbstractPhil/imagenet-clip-features-orderly",
"model_variant": "clip_vit_laion_b32",
"num_epochs": 10,
"batch_size": 1024,
"learning_rate": 0.001,
"rose_weight": "0.1\u21920.5",
"cayley_loss": false,
"optimizer": "AdamW",
"scheduler": "cosine_restarts"
},
"files": {
"weights_safetensors": "weights/David-hierarchical-progressive/20251012_145649/best_model_acc76.81.safetensors",
"weights_pytorch": "weights/David-hierarchical-progressive/20251012_145649/best_model.pth",
"config": "weights/David-hierarchical-progressive/20251012_145649/david_config.json",
"training_config": "weights/David-hierarchical-progressive/20251012_145649/train_config.json",
"tensorboard": "runs/David-hierarchical-progressive/20251012_145649/"
}
}