Update beatrix-cifar100_20251007_195812 - Epoch 0 - Acc: 0.0492

README.md

@@ -16,13 +16,14 @@ with geometric simplex features for vision tasks.
 
 ## Available Models (Best Checkpoints Only)
 
-| Model Name | Training Session | Accuracy | Epoch | Path |
-|------------|------------------|----------|-------|------|
-| beatrix-cifar100 | 20251007_182851 | 0.5819 | 42 | `weights/beatrix-cifar100/20251007_182851` |
-| beatrix-cifar100 | 20251007_193112 | 0.2802 | 10 | `weights/beatrix-cifar100/20251007_193112` |
+| Model Name | Training Session | Accuracy | Epoch | Weights Path | Logs Path |
+|------------|------------------|----------|-------|--------------|----------|
+| beatrix-cifar100 | 20251007_182851 | 0.5819 | 42 | `weights/beatrix-cifar100/20251007_182851` | `N/A` |
+| beatrix-cifar100 | 20251007_193112 | 0.2802 | 10 | `weights/beatrix-cifar100/20251007_193112` | `N/A` |
+| beatrix-cifar100 | 20251007_195812 | 0.0492 | 0 | `weights/beatrix-cifar100/20251007_195812` | `logs/beatrix-cifar100/20251007_195812` |
 
 
-## Latest Updated Model: beatrix-cifar100 (Session: 20251007_193112)
+## Latest Updated Model: beatrix-cifar100 (Session: 20251007_195812)
 
 ### Model Details
 
@@ -36,9 +37,9 @@ with geometric simplex features for vision tasks.
 
 ### Training Details
 
-- **Training Session**: 20251007_193112
-- **Best Accuracy**: 0.2802
-- **Epochs Trained**: 10
+- **Training Session**: 20251007_195812
+- **Best Accuracy**: 0.0492
+- **Epochs Trained**: 0
 - **Batch Size**: 512
 - **Learning Rate**: 0.0001
 
@@ -49,6 +50,22 @@ with geometric simplex features for vision tasks.
 - Coherence Weight: 0.3
 - Multi-Scale Weight: 0.2
 
+### TensorBoard Logs
+
+Training logs are available in the repository at:
+```
+logs/beatrix-cifar100/20251007_195812
+```
+
+To view them locally:
+```bash
+# Clone the repo
+git clone https://huggingface.co/AbstractPhil/vit-beatrix
+
+# View logs in TensorBoard
+tensorboard --logdir vit-beatrix/logs/beatrix-cifar100/20251007_195812
+```
+
 ## Usage
 
 ```python
@@ -73,7 +90,7 @@ for key, info in sorted(manifest.items(), key=lambda x: x[1]['accuracy'], revers
 # Download weights for the latest training session of beatrix-cifar100
 weights_path = hf_hub_download(
     repo_id="AbstractPhil/vit-beatrix",
-    filename="weights/beatrix-cifar100/20251007_193112/model.safetensors"
+    filename="weights/beatrix-cifar100/20251007_195812/model.safetensors"
 )
 
 # Load model

logs/beatrix-cifar100/20251007_195812/events.out.tfevents.1759867094.f41d735e1a7f.28489.3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fdc505a2bb96edfca1fca6d4762f04ca163742d7e7709d8583c7afbaa1f0ad07
+size 88

manifest.json

@@ -19,5 +19,17 @@
     "depth": 12,
     "batch_size": 512,
     "learning_rate": 0.0001
+  },
+  "beatrix-cifar100_20251007_195812": {
+    "accuracy": 0.04920726120471954,
+    "epoch": 0,
+    "timestamp": "20251007_195812",
+    "model_name": "beatrix-cifar100",
+    "path": "weights/beatrix-cifar100/20251007_195812",
+    "logs_path": "logs/beatrix-cifar100/20251007_195812",
+    "embed_dim": 256,
+    "depth": 12,
+    "batch_size": 512,
+    "learning_rate": 0.0001
   }
 }

weights/beatrix-cifar100/20251007_195812/config.json

@@ -0,0 +1,45 @@
+{
+  "num_classes": 100,
+  "img_size": 32,
+  "patch_size": 4,
+  "embed_dim": 256,
+  "k_simplex": 4,
+  "depth": 12,
+  "num_heads": 8,
+  "mlp_ratio": 4.0,
+  "dropout": 0.0,
+  "pe_levels": 12,
+  "pe_features_per_level": 2,
+  "pe_smooth_tau": 0.25,
+  "simplex_feature_weight": 0.5,
+  "batch_size": 512,
+  "num_epochs": 50,
+  "learning_rate": 0.0001,
+  "weight_decay": 0.005,
+  "warmup_epochs": 10,
+  "task_loss_weight": 0.5,
+  "flow_loss_weight": 0.5,
+  "coherence_loss_weight": 0.3,
+  "multiscale_loss_weight": 0.2,
+  "volume_reg_weight": 0.1,
+  "use_transformed_augmentation": true,
+  "use_augmentation": false,
+  "mixup_alpha": 0.2,
+  "cutmix_alpha": 1.0,
+  "device": "cuda",
+  "num_workers": 4,
+  "pin_memory": true,
+  "save_dir": "./checkpoints",
+  "save_every": 10,
+  "use_safetensors": true,
+  "timestamp_dirs": true,
+  "push_to_hub": true,
+  "hub_model_id": "AbstractPhil/vit-beatrix",
+  "hub_model_name": "beatrix-cifar100",
+  "hub_upload_best_only": true,
+  "hub_upload_every_n_epochs": 10,
+  "use_tensorboard": true,
+  "log_dir": "./logs",
+  "log_every": 50,
+  "train_baseline": false
+}

weights/beatrix-cifar100/20251007_195812/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dad540b27e862051356c8653d2f221b08ee1eb8afe0653fe931ccceccb13851c
+size 38116128