AbstractPhil commited on
Commit
81834f9
·
verified ·
1 Parent(s): 06529d6

Update metrics - Run 20251012_191456

Browse files
Files changed (1) hide show
  1. best_model.json +48 -0
best_model.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "David-partial_shared-hierarchical_tree",
3
+ "run_id": "20251012_191456",
4
+ "timestamp": "2025-10-12T19:18:12.460521",
5
+ "best_val_acc": 66.689,
6
+ "best_epoch": 0,
7
+ "final_train_acc": 54.52540535308824,
8
+ "final_train_loss": 3.4310503170769358,
9
+ "scale_accuracies": {
10
+ "256": 66.689
11
+ },
12
+ "architecture": {
13
+ "preset": "balanced",
14
+ "sharing_mode": "partial_shared",
15
+ "fusion_mode": "hierarchical_tree",
16
+ "scales": [
17
+ 256,
18
+ 512,
19
+ 768,
20
+ 1024
21
+ ],
22
+ "feature_dim": 512,
23
+ "num_classes": 1000,
24
+ "use_belly": true,
25
+ "belly_expand": 2.0
26
+ },
27
+ "training": {
28
+ "dataset": "AbstractPhil/imagenet-clip-features-orderly",
29
+ "model_variant": [
30
+ "clip_vit_b32",
31
+ "clip_vit_laion_b32"
32
+ ],
33
+ "num_epochs": 10,
34
+ "batch_size": 1024,
35
+ "learning_rate": 0.01,
36
+ "rose_weight": "0.2\u21920.8",
37
+ "cayley_loss": true,
38
+ "optimizer": "AdamW",
39
+ "scheduler": "cosine_restarts"
40
+ },
41
+ "files": {
42
+ "weights_safetensors": "weights/David-partial_shared-hierarchical_tree/20251012_191456/best_model_acc66.69.safetensors",
43
+ "weights_pytorch": "weights/David-partial_shared-hierarchical_tree/20251012_191456/best_model.pth",
44
+ "config": "weights/David-partial_shared-hierarchical_tree/20251012_191456/david_config.json",
45
+ "training_config": "weights/David-partial_shared-hierarchical_tree/20251012_191456/train_config.json",
46
+ "tensorboard": "runs/David-partial_shared-hierarchical_tree/20251012_191456/"
47
+ }
48
+ }