AbstractPhil commited on
Commit
6d1c8ca
·
verified ·
1 Parent(s): 2d7cc6d

Update beatrix-dualstream-base (Epoch 0, Acc: 0.0395)

Browse files
README.md CHANGED
@@ -13,15 +13,15 @@ license: mit
13
 
14
  ## Current Experiment: beatrix-dualstream-base
15
 
16
- **Model Path**: `weights/beatrix-dualstream-base/20251009_025739/`
17
 
18
  ## Architecture
19
 
20
- - **Visual Dimension**: 128
21
- - **Geometric Dimension**: 512
22
- - **Geometric Tokens**: 16
23
  - **Dual Blocks**: 8 layers
24
- - **k-simplex**: 12
25
 
26
  ## Geometric Diversity Loss
27
 
@@ -33,10 +33,10 @@ This model uses a class-aware geometric diversity loss that encourages:
33
 
34
  ## Performance
35
 
36
- - **Best Accuracy**: 0.0254
37
  - **Current Epoch**: 0
38
  - **Dataset**: CIFAR-100
39
 
40
  ---
41
 
42
- *Last updated: Epoch 0 | Best Accuracy: 0.0254*
 
13
 
14
  ## Current Experiment: beatrix-dualstream-base
15
 
16
+ **Model Path**: `weights/beatrix-dualstream-base/20251009_030219/`
17
 
18
  ## Architecture
19
 
20
+ - **Visual Dimension**: 768
21
+ - **Geometric Dimension**: 768
22
+ - **Geometric Tokens**: 32
23
  - **Dual Blocks**: 8 layers
24
+ - **k-simplex**: 9
25
 
26
  ## Geometric Diversity Loss
27
 
 
33
 
34
  ## Performance
35
 
36
+ - **Best Accuracy**: 0.0395
37
  - **Current Epoch**: 0
38
  - **Dataset**: CIFAR-100
39
 
40
  ---
41
 
42
+ *Last updated: Epoch 0 | Best Accuracy: 0.0395*
weights/beatrix-dualstream-base/20251009_030219/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "num_classes": 100,
3
+ "img_size": 32,
4
+ "patch_size": 8,
5
+ "visual_dim": 768,
6
+ "geom_dim": 768,
7
+ "k_simplex": 9,
8
+ "depth": 8,
9
+ "num_heads": 8,
10
+ "mlp_ratio": 4.0,
11
+ "dropout": 0.0,
12
+ "num_geom_tokens": 32,
13
+ "classification_goal": "both",
14
+ "pe_levels": 24,
15
+ "pe_features_per_level": 2,
16
+ "pe_smooth_tau": 0.25,
17
+ "simplex_init_method": "regular",
18
+ "simplex_init_scale": 1.0,
19
+ "batch_size": 1024,
20
+ "num_epochs": 100,
21
+ "learning_rate": 0.0001,
22
+ "weight_decay": 0.005,
23
+ "warmup_epochs": 10,
24
+ "task_loss_weight": 0.5,
25
+ "flow_loss_weight": 1.0,
26
+ "coherence_loss_weight": 0.3,
27
+ "multiscale_loss_weight": 0.2,
28
+ "diversity_loss_weight": 0.5,
29
+ "diversity_margin": 2.0,
30
+ "diversity_temperature": 0.1,
31
+ "diversity_intra_weight": 1.0,
32
+ "diversity_inter_weight": 2.0,
33
+ "diversity_volume_weight": 0.5,
34
+ "diversity_complexity_weight": 0.1,
35
+ "device": "cuda",
36
+ "num_workers": 4,
37
+ "pin_memory": true,
38
+ "save_dir": "./checkpoints_dualstream",
39
+ "save_every": 10,
40
+ "use_safetensors": true,
41
+ "timestamp_dirs": true,
42
+ "push_to_hub": true,
43
+ "hub_model_id": "AbstractPhil/vit-beatrix-dualstream",
44
+ "hub_model_name": "beatrix-dualstream-base",
45
+ "hub_upload_best_only": true,
46
+ "hub_upload_every_n_epochs": 10,
47
+ "use_tensorboard": true,
48
+ "log_dir": "./logs_dualstream",
49
+ "log_every": 50,
50
+ "monitor_stream_health": true,
51
+ "log_stream_norms": true
52
+ }
weights/beatrix-dualstream-base/20251009_030219/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16153f5bb6567ab1708528a49de4e806e066cc8c1b1541e8293e1204c3cd6aaa
3
+ size 606483468