gated-david / best_model.json
AbstractPhil's picture
Update metrics - Run 20251104_152832
c46d88f verified
raw
history blame
1.45 kB
{
"model_name": "David-decoupled-cantor_scale",
"run_id": "20251104_152832",
"timestamp": "2025-11-04T15:39:28.692091",
"best_val_acc": 76.498,
"best_epoch": 4,
"final_train_acc": 82.7423747255432,
"final_train_loss": 1.913906869438711,
"scale_accuracies": {
"256": 71.268,
"512": 74.69,
"768": 75.028,
"1024": 75.528
},
"architecture": {
"preset": "clip_vit_b16_cantor",
"sharing_mode": "decoupled",
"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_laion_b32",
"num_epochs": 5,
"batch_size": 512,
"learning_rate": 0.001,
"rose_weight": "0.1\u21920.5",
"cayley_loss": false,
"optimizer": "AdamW",
"scheduler": "cosine_restarts"
},
"files": {
"weights_safetensors": "weights/David-decoupled-cantor_scale/20251104_152832/best_model_acc76.50.safetensors",
"weights_pytorch": "weights/David-decoupled-cantor_scale/20251104_152832/best_model.pth",
"config": "weights/David-decoupled-cantor_scale/20251104_152832/david_config.json",
"training_config": "weights/David-decoupled-cantor_scale/20251104_152832/train_config.json",
"tensorboard": "runs/David-decoupled-cantor_scale/20251104_152832/"
}
}