Update beatrix-cifar100_20251007_203741 - Epoch 0 - Acc: 0.0470
Browse files
README.md
CHANGED
|
@@ -14,6 +14,28 @@ license: mit
|
|
| 14 |
This repository contains Vision Transformers integrating Devil's Staircase positional encoding
|
| 15 |
with geometric simplex features for vision tasks.
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
## Available Models (Best Checkpoints Only)
|
| 18 |
|
| 19 |
| Model Name | Training Session | Accuracy | Epoch | Weights Path | Logs Path |
|
|
@@ -21,9 +43,10 @@ with geometric simplex features for vision tasks.
|
|
| 21 |
| beatrix-cifar100 | 20251007_182851 | 0.5819 | 42 | `weights/beatrix-cifar100/20251007_182851` | `N/A` |
|
| 22 |
| beatrix-cifar100 | 20251007_195812 | 0.4701 | 42 | `weights/beatrix-cifar100/20251007_195812` | `logs/beatrix-cifar100/20251007_195812` |
|
| 23 |
| beatrix-cifar100 | 20251007_193112 | 0.2802 | 10 | `weights/beatrix-cifar100/20251007_193112` | `N/A` |
|
|
|
|
| 24 |
|
| 25 |
|
| 26 |
-
## Latest Updated Model: beatrix-cifar100 (Session:
|
| 27 |
|
| 28 |
### Model Details
|
| 29 |
|
|
@@ -37,11 +60,12 @@ with geometric simplex features for vision tasks.
|
|
| 37 |
|
| 38 |
### Training Details
|
| 39 |
|
| 40 |
-
- **Training Session**:
|
| 41 |
-
- **Best Accuracy**: 0.
|
| 42 |
-
- **Epochs Trained**:
|
| 43 |
- **Batch Size**: 512
|
| 44 |
- **Learning Rate**: 0.0001
|
|
|
|
| 45 |
|
| 46 |
### Loss Configuration
|
| 47 |
|
|
@@ -54,7 +78,7 @@ with geometric simplex features for vision tasks.
|
|
| 54 |
|
| 55 |
Training logs are available in the repository at:
|
| 56 |
```
|
| 57 |
-
logs/beatrix-cifar100/
|
| 58 |
```
|
| 59 |
|
| 60 |
To view them locally:
|
|
@@ -63,11 +87,32 @@ To view them locally:
|
|
| 63 |
git clone https://huggingface.co/AbstractPhil/vit-beatrix
|
| 64 |
|
| 65 |
# View logs in TensorBoard
|
| 66 |
-
tensorboard --logdir vit-beatrix/logs/beatrix-cifar100/
|
| 67 |
```
|
| 68 |
|
| 69 |
## Usage
|
| 70 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
```python
|
| 72 |
from geovocab2.train.model.vit_beatrix import SimplifiedGeometricClassifier
|
| 73 |
from safetensors.torch import load_file
|
|
@@ -90,7 +135,7 @@ for key, info in sorted(manifest.items(), key=lambda x: x[1]['accuracy'], revers
|
|
| 90 |
# Download weights for the latest training session of beatrix-cifar100
|
| 91 |
weights_path = hf_hub_download(
|
| 92 |
repo_id="AbstractPhil/vit-beatrix",
|
| 93 |
-
filename="weights/beatrix-cifar100/
|
| 94 |
)
|
| 95 |
|
| 96 |
# Load model
|
|
|
|
| 14 |
This repository contains Vision Transformers integrating Devil's Staircase positional encoding
|
| 15 |
with geometric simplex features for vision tasks.
|
| 16 |
|
| 17 |
+
## Key Features
|
| 18 |
+
|
| 19 |
+
- **Fractal Positional Encoding**: Devil's Staircase multi-scale position embeddings
|
| 20 |
+
- **Geometric Simplex Features**: k-simplex vertex computations from Cantor measure
|
| 21 |
+
- **Adaptive Augmentation**: Progressive augmentation escalation to prevent overfitting
|
| 22 |
+
- **Beatrix Formula Suite**: Flow alignment, hierarchical coherence, and multi-scale consistency losses
|
| 23 |
+
|
| 24 |
+
### Adaptive Augmentation System
|
| 25 |
+
|
| 26 |
+
The trainer includes an intelligent augmentation system that monitors train/validation accuracy gap and progressively enables more augmentation:
|
| 27 |
+
|
| 28 |
+
1. **Baseline**: RandomCrop + RandomHorizontalFlip
|
| 29 |
+
2. **Stage 1**: + ColorJitter
|
| 30 |
+
3. **Stage 2**: + RandomRotation
|
| 31 |
+
4. **Stage 3**: + RandomAffine
|
| 32 |
+
5. **Stage 4**: + RandomErasing
|
| 33 |
+
6. **Stage 5**: + AutoAugment (CIFAR policy)
|
| 34 |
+
7. **Stage 6**: Enable Mixup (α=0.2)
|
| 35 |
+
8. **Stage 7**: Enable CutMix (α=1.0) - Final stage
|
| 36 |
+
|
| 37 |
+
When train accuracy exceeds validation accuracy by 2% or more, the system automatically escalates to the next augmentation stage.
|
| 38 |
+
|
| 39 |
## Available Models (Best Checkpoints Only)
|
| 40 |
|
| 41 |
| Model Name | Training Session | Accuracy | Epoch | Weights Path | Logs Path |
|
|
|
|
| 43 |
| beatrix-cifar100 | 20251007_182851 | 0.5819 | 42 | `weights/beatrix-cifar100/20251007_182851` | `N/A` |
|
| 44 |
| beatrix-cifar100 | 20251007_195812 | 0.4701 | 42 | `weights/beatrix-cifar100/20251007_195812` | `logs/beatrix-cifar100/20251007_195812` |
|
| 45 |
| beatrix-cifar100 | 20251007_193112 | 0.2802 | 10 | `weights/beatrix-cifar100/20251007_193112` | `N/A` |
|
| 46 |
+
| beatrix-cifar100 | 20251007_203741 | 0.0470 | 0 | `weights/beatrix-cifar100/20251007_203741` | `logs/beatrix-cifar100/20251007_203741` |
|
| 47 |
|
| 48 |
|
| 49 |
+
## Latest Updated Model: beatrix-cifar100 (Session: 20251007_203741)
|
| 50 |
|
| 51 |
### Model Details
|
| 52 |
|
|
|
|
| 60 |
|
| 61 |
### Training Details
|
| 62 |
|
| 63 |
+
- **Training Session**: 20251007_203741
|
| 64 |
+
- **Best Accuracy**: 0.0470
|
| 65 |
+
- **Epochs Trained**: 0
|
| 66 |
- **Batch Size**: 512
|
| 67 |
- **Learning Rate**: 0.0001
|
| 68 |
+
- **Adaptive Augmentation**: Enabled
|
| 69 |
|
| 70 |
### Loss Configuration
|
| 71 |
|
|
|
|
| 78 |
|
| 79 |
Training logs are available in the repository at:
|
| 80 |
```
|
| 81 |
+
logs/beatrix-cifar100/20251007_203741
|
| 82 |
```
|
| 83 |
|
| 84 |
To view them locally:
|
|
|
|
| 87 |
git clone https://huggingface.co/AbstractPhil/vit-beatrix
|
| 88 |
|
| 89 |
# View logs in TensorBoard
|
| 90 |
+
tensorboard --logdir vit-beatrix/logs/beatrix-cifar100/20251007_203741
|
| 91 |
```
|
| 92 |
|
| 93 |
## Usage
|
| 94 |
|
| 95 |
+
### Installation
|
| 96 |
+
|
| 97 |
+
For Google Colab:
|
| 98 |
+
```python
|
| 99 |
+
# Install for Colab
|
| 100 |
+
try:
|
| 101 |
+
!pip uninstall -qy geometricvocab
|
| 102 |
+
except:
|
| 103 |
+
pass
|
| 104 |
+
|
| 105 |
+
!pip install -q git+https://github.com/AbstractEyes/lattice_vocabulary.git
|
| 106 |
+
```
|
| 107 |
+
|
| 108 |
+
For local environments:
|
| 109 |
+
```bash
|
| 110 |
+
# install the repo into your environment
|
| 111 |
+
pip install -q git+https://github.com/AbstractEyes/lattice_vocabulary.git
|
| 112 |
+
```
|
| 113 |
+
|
| 114 |
+
### Loading Models
|
| 115 |
+
|
| 116 |
```python
|
| 117 |
from geovocab2.train.model.vit_beatrix import SimplifiedGeometricClassifier
|
| 118 |
from safetensors.torch import load_file
|
|
|
|
| 135 |
# Download weights for the latest training session of beatrix-cifar100
|
| 136 |
weights_path = hf_hub_download(
|
| 137 |
repo_id="AbstractPhil/vit-beatrix",
|
| 138 |
+
filename="weights/beatrix-cifar100/20251007_203741/model.safetensors"
|
| 139 |
)
|
| 140 |
|
| 141 |
# Load model
|
logs/beatrix-cifar100/20251007_203741/events.out.tfevents.1759869463.f41d735e1a7f.65607.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7ca3589705bba9a0c2248a76818c50adfacecc9ef7a63f9a9539c3babe019b84
|
| 3 |
+
size 88
|
manifest.json
CHANGED
|
@@ -31,5 +31,17 @@
|
|
| 31 |
"depth": 12,
|
| 32 |
"batch_size": 512,
|
| 33 |
"learning_rate": 0.0001
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
}
|
| 35 |
}
|
|
|
|
| 31 |
"depth": 12,
|
| 32 |
"batch_size": 512,
|
| 33 |
"learning_rate": 0.0001
|
| 34 |
+
},
|
| 35 |
+
"beatrix-cifar100_20251007_203741": {
|
| 36 |
+
"accuracy": 0.04703010115772486,
|
| 37 |
+
"epoch": 0,
|
| 38 |
+
"timestamp": "20251007_203741",
|
| 39 |
+
"model_name": "beatrix-cifar100",
|
| 40 |
+
"path": "weights/beatrix-cifar100/20251007_203741",
|
| 41 |
+
"logs_path": "logs/beatrix-cifar100/20251007_203741",
|
| 42 |
+
"embed_dim": 256,
|
| 43 |
+
"depth": 12,
|
| 44 |
+
"batch_size": 512,
|
| 45 |
+
"learning_rate": 0.0001
|
| 46 |
}
|
| 47 |
}
|
weights/beatrix-cifar100/20251007_203741/config.json
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"num_classes": 100,
|
| 3 |
+
"img_size": 32,
|
| 4 |
+
"patch_size": 4,
|
| 5 |
+
"embed_dim": 256,
|
| 6 |
+
"k_simplex": 4,
|
| 7 |
+
"depth": 12,
|
| 8 |
+
"num_heads": 8,
|
| 9 |
+
"mlp_ratio": 4.0,
|
| 10 |
+
"dropout": 0.0,
|
| 11 |
+
"pe_levels": 12,
|
| 12 |
+
"pe_features_per_level": 2,
|
| 13 |
+
"pe_smooth_tau": 0.25,
|
| 14 |
+
"simplex_feature_weight": 0.5,
|
| 15 |
+
"batch_size": 512,
|
| 16 |
+
"num_epochs": 50,
|
| 17 |
+
"learning_rate": 0.0001,
|
| 18 |
+
"weight_decay": 0.005,
|
| 19 |
+
"warmup_epochs": 10,
|
| 20 |
+
"task_loss_weight": 0.5,
|
| 21 |
+
"flow_loss_weight": 0.5,
|
| 22 |
+
"coherence_loss_weight": 0.3,
|
| 23 |
+
"multiscale_loss_weight": 0.2,
|
| 24 |
+
"volume_reg_weight": 0.1,
|
| 25 |
+
"use_adaptive_augmentation": true,
|
| 26 |
+
"overfit_threshold": 0.02,
|
| 27 |
+
"mixup_alpha": 0.2,
|
| 28 |
+
"cutmix_alpha": 1.0,
|
| 29 |
+
"device": "cuda",
|
| 30 |
+
"num_workers": 4,
|
| 31 |
+
"pin_memory": true,
|
| 32 |
+
"save_dir": "./checkpoints",
|
| 33 |
+
"save_every": 10,
|
| 34 |
+
"use_safetensors": true,
|
| 35 |
+
"timestamp_dirs": true,
|
| 36 |
+
"push_to_hub": true,
|
| 37 |
+
"hub_model_id": "AbstractPhil/vit-beatrix",
|
| 38 |
+
"hub_model_name": "beatrix-cifar100",
|
| 39 |
+
"hub_upload_best_only": true,
|
| 40 |
+
"hub_upload_every_n_epochs": 10,
|
| 41 |
+
"use_tensorboard": true,
|
| 42 |
+
"log_dir": "./logs",
|
| 43 |
+
"log_every": 50,
|
| 44 |
+
"train_baseline": false
|
| 45 |
+
}
|
weights/beatrix-cifar100/20251007_203741/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e46631fad61dd7418830b1b01b1266024244b295b08853e8a7098ea683b91f6d
|
| 3 |
+
size 38116128
|