gated-david / best_model.json
AbstractPhil's picture
Update metrics - Run 20251104_124032
5563dbe verified
raw
history blame
1.48 kB
{
"model_name": "David-partial_shared-cantor_scale",
"run_id": "20251104_124032",
"timestamp": "2025-11-04T12:50:33.273837",
"best_val_acc": 78.182,
"best_epoch": 4,
"final_train_acc": 84.90430989870954,
"final_train_loss": 1.570512046543554,
"scale_accuracies": {
"256": 76.696,
"512": 77.532,
"768": 77.624,
"1024": 77.56
},
"architecture": {
"preset": "clip_vit_b16_cantor",
"sharing_mode": "partial_shared",
"fusion_mode": "cantor_scale",
"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_b16",
"num_epochs": 5,
"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-partial_shared-cantor_scale/20251104_124032/best_model_acc78.18.safetensors",
"weights_pytorch": "weights/David-partial_shared-cantor_scale/20251104_124032/best_model.pth",
"config": "weights/David-partial_shared-cantor_scale/20251104_124032/david_config.json",
"training_config": "weights/David-partial_shared-cantor_scale/20251104_124032/train_config.json",
"tensorboard": "runs/David-partial_shared-cantor_scale/20251104_124032/"
}
}