AbstractPhil commited on
Commit
6c4ab27
·
verified ·
1 Parent(s): 87bb52f

Epoch 10 - 44.60% acc

Browse files
weights/run_003_v2_8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix_20251202_142542/README.md ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Run: run_003_v2_8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix_20251202_142542
2
+
3
+ ## Results
4
+ - **Best Accuracy**: 44.60%
5
+ - **Dataset**: cifar100
6
+ - **Epochs**: 120
7
+ - **Model Version**: V2
8
+
9
+ ## Model Config
10
+ | Parameter | Value |
11
+ |-----------|-------|
12
+ | Dim | 512 |
13
+ | Layers | 4 |
14
+ | Heads | 16 |
15
+ | Patch Size | 4 |
16
+
17
+ ## Wormhole Routing (V2)
18
+ | Parameter | Value |
19
+ |-----------|-------|
20
+ | Mode | hybrid |
21
+ | Wormholes/Position | 8 |
22
+ | Temperature | 0.1 |
23
+ | Tiles | 16 |
24
+ | Tile Wormholes | 4 |
25
+
26
+ ## Crystal Head
27
+ | Parameter | Value |
28
+ |-----------|-------|
29
+ | Scales | [64, 128, 256, 512] |
30
+ | Scale Copies | [8, 4, 2, 2] |
31
+ | Weighting Mode | geometric |
32
+ | Belly Layers | 2 |
33
+ | Belly Residual | False |
34
+ | Use Spine | True |
35
+ | Use Collective | False |
36
+
37
+
38
+ ## Training Config
39
+ | Parameter | Value |
40
+ |-----------|-------|
41
+ | Learning Rate | 0.0003 |
42
+ | Weight Decay | 0.05 |
43
+ | Batch Size | 128 |
44
+ | CE Weight | 1.0 |
45
+ | Contrast Weight | 0.5 |
46
+
47
+ ## Augmentation
48
+ | Parameter | Value |
49
+ |-----------|-------|
50
+ | Normalization | standard |
51
+ | Mixup Alpha | 0.0 |
52
+ | CutMix Alpha | 0.0 |
53
+ | AlphaMix | False |
54
+ | Label Smoothing | 0.1 |
55
+
56
+
57
+ ## Key Findings Applied
58
+ - Routing learns from task pressure (no auxiliary routing losses)
59
+ - Gradients verified to flow through router
60
+ - Cross-contrastive aligns patch↔scale features
weights/run_003_v2_8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix_20251202_142542/best.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2a2e78a5777dee813459822c3537ce260c49345ae948f6fc4635652db381ee0
3
+ size 98300616
weights/run_003_v2_8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix_20251202_142542/config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "DavidBeans_V2",
3
+ "model_type": "david_beans_v2",
4
+ "image_size": 32,
5
+ "patch_size": 4,
6
+ "in_channels": 3,
7
+ "dim": 512,
8
+ "num_layers": 4,
9
+ "num_heads": 16,
10
+ "mlp_ratio": 4.0,
11
+ "num_wormholes": 8,
12
+ "wormhole_temperature": 0.1,
13
+ "wormhole_mode": "hybrid",
14
+ "cantor_weight": 0.3,
15
+ "num_tiles": 16,
16
+ "tile_wormholes": 4,
17
+ "scales": [
18
+ 64,
19
+ 128,
20
+ 256,
21
+ 512
22
+ ],
23
+ "num_classes": 100,
24
+ "use_belly": true,
25
+ "belly_expand": 2.0,
26
+ "belly_layers": 2,
27
+ "belly_residual": false,
28
+ "weighting_mode": "geometric",
29
+ "scale_weight_floor": 0.1,
30
+ "scale_copies": [
31
+ 8,
32
+ 4,
33
+ 2,
34
+ 2
35
+ ],
36
+ "copy_theta_step": 0.15,
37
+ "use_collective": false,
38
+ "collective_temperature": 0.07,
39
+ "use_spine": true,
40
+ "spine_channels": [
41
+ 64,
42
+ 128,
43
+ 256
44
+ ],
45
+ "spine_cross_attn": true,
46
+ "spine_gate_init": 0.0,
47
+ "contrast_temperature": 0.07,
48
+ "contrast_weight": 0.5,
49
+ "routing_weight": 0.0,
50
+ "dropout": 0.1,
51
+ "use_topo_dropout": true,
52
+ "topo_drop_prob": 0.15,
53
+ "topo_warmup_epochs": 35,
54
+ "topo_min_routes_keep": 2,
55
+ "topo_steps_per_epoch": 391,
56
+ "use_spatial_dropout": true,
57
+ "spatial_drop_prob": 0.1,
58
+ "spatial_patch_size": 4,
59
+ "route_dim_attention": -2,
60
+ "route_dim_tiles": 2,
61
+ "pooling": "cls"
62
+ }
weights/run_003_v2_8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix_20251202_142542/tensorboard/events.out.tfevents.1764685544.7ccc73cc3f03.364.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06e189f7e84f2e0be759f5128749a6e2210eaec4f2a96971c19426f6442613fa
3
+ size 36912
weights/run_003_v2_8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix_20251202_142542/training_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "run_name": "8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix",
3
+ "run_number": null,
4
+ "model_version": 2,
5
+ "dataset": "cifar100",
6
+ "image_size": 32,
7
+ "batch_size": 128,
8
+ "num_workers": 4,
9
+ "normalization": "standard",
10
+ "epochs": 120,
11
+ "warmup_epochs": 10,
12
+ "learning_rate": 0.0003,
13
+ "weight_decay": 0.05,
14
+ "betas": [
15
+ 0.9,
16
+ 0.999
17
+ ],
18
+ "scheduler": "cosine",
19
+ "min_lr": 1e-06,
20
+ "ce_weight": 1.0,
21
+ "contrast_weight": 0.5,
22
+ "gradient_clip": 1.0,
23
+ "label_smoothing": 0.1,
24
+ "use_augmentation": true,
25
+ "mixup_alpha": 0.0,
26
+ "cutmix_alpha": 0.0,
27
+ "use_alphamix": false,
28
+ "alphamix_alpha_range": [
29
+ 0.3,
30
+ 0.7
31
+ ],
32
+ "alphamix_spatial_ratio": 0.25,
33
+ "save_interval": 10,
34
+ "output_dir": "./checkpoints/cifar100_v2",
35
+ "resume_from": null,
36
+ "use_tensorboard": true,
37
+ "log_interval": 50,
38
+ "log_routing": true,
39
+ "push_to_hub": true,
40
+ "hub_repo_id": "AbstractPhil/geovit-david-beans",
41
+ "hub_private": false,
42
+ "device": "cuda"
43
+ }
weights/run_003_v2_8redundantscale_4x4patch_d768_4layer_topo_dropout_cutmix_20251202_142542/training_history.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "loss": [
3
+ 13.873117962861672,
4
+ 12.275092814518855,
5
+ 11.182369144146259,
6
+ 10.339609992198454,
7
+ 9.659815986339863,
8
+ 9.001137947424864,
9
+ 8.437557945495996,
10
+ 8.010158390876574,
11
+ 7.641950374994522
12
+ ],
13
+ "ce": [
14
+ 4.3127842603585655,
15
+ 3.8434408181752913,
16
+ 3.505825737806467,
17
+ 3.2513973236083986,
18
+ 3.0415021242239537,
19
+ 2.835707812431531,
20
+ 2.657775013263409,
21
+ 2.5238647595430033,
22
+ 2.408466780797029
23
+ ],
24
+ "contrast": [
25
+ 4.463720106467223,
26
+ 3.955229284824469,
27
+ 3.6059489427468714,
28
+ 3.3343552149259126,
29
+ 3.1169237479185448,
30
+ 2.90509451230367,
31
+ 2.724379973533826,
32
+ 2.5893612372569548,
33
+ 2.471492732488192
34
+ ],
35
+ "lr": [
36
+ 2.9999999999999997e-05,
37
+ 5.9999999999999995e-05,
38
+ 8.999999999999999e-05,
39
+ 0.00011999999999999999,
40
+ 0.00015,
41
+ 0.00017999999999999998,
42
+ 0.00020999999999999995,
43
+ 0.00023999999999999998,
44
+ 0.00026999999999999995
45
+ ]
46
+ }