gated-david / best_model.json
AbstractPhil's picture
Update metrics - Run 20251012_145649
6a42fac verified
raw
history blame
1.47 kB
{
"model_name": "David-hierarchical-progressive",
"run_id": "20251012_145649",
"timestamp": "2025-10-12T15:12:51.543853",
"best_val_acc": 76.862,
"best_epoch": 8,
"final_train_acc": 90.4128813808036,
"final_train_loss": 1.4593032695614874,
"scale_accuracies": {
"256": 70.244,
"512": 74.392,
"768": 75.64,
"1024": 75.514
},
"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.86.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/"
}
}