Update model - STAGE2 Epoch 1 | Loss: 5.2714

README.md

@@ -10,7 +10,7 @@ tags:
 - tiny-vlm
 - repvit
 - tinyllm
-- stage1
+- stage2
 base_model:
 - tinyllm
 library_name: transformers
@@ -21,18 +21,17 @@ pipeline_tag: image-text-to-text
 
 **🔥 Efficient Vision-Language Model for Edge Deployment & Robotic Applications**
 
-This model is currently in training - **STAGE1 (Epoch 1)**.
+This model is currently in training - **STAGE2 (Epoch 1)**.
 
 ## 📊 Current Training Status
 
-- **Stage**: Visual-Language Alignment - Learning to ground vision and language
+- **Stage**: Multimodal Instruction Tuning - Following complex instructions
 - **Epoch**: 1
-- **Last Updated**: 2026-02-01 16:00:11 UTC
+- **Last Updated**: 2026-02-01 16:01:18 UTC
 
 ### Latest Metrics
-- **captioning_loss**: 8.5561
-- **contrastive_loss**: 2.7994
-- **loss**: 5.6777
+- **instruction_loss**: 0.0000
+- **loss**: 5.2714
 
 ## 🏗️ Model Architecture
 
@@ -51,7 +50,7 @@ EmberVLM follows a 4-stage training curriculum:
 3. ✅ **Stage 3: Robot Fleet Selection** - Task-robot matching
 4. ⏳ **Stage 4: Chain-of-Thought Reasoning** - Reasoning generation
 
-**Current Stage**: STAGE1
+**Current Stage**: STAGE2
 
 ## 💻 Usage
 
@@ -126,5 +125,5 @@ Apache 2.0
 
 ---
 
-**Note**: This is a checkpoint from stage1 training (epoch 1). 
+**Note**: This is a checkpoint from stage2 training (epoch 1). 
 The model will be updated after each epoch with improved performance.

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c6be11d39bd7c475a6e51883249a0a9ba175c11618e424678674cb2ef649fe66
+oid sha256:471ac551d9b840a27745efe6b15526884be81d959b0082c65c1f5e2ba9f5bd97
 size 100663623

training_info.json

@@ -1,13 +1,12 @@
 {
-  "stage": "stage1",
+  "stage": "stage2",
   "epoch": 1,
   "metrics": {
-    "loss": 5.6777140368586005,
-    "contrastive_loss": 2.7993588654891304,
-    "captioning_loss": 8.556068959443465
+    "loss": 5.271356953514947,
+    "instruction_loss": 0.0
   },
   "carbon_emissions_kg": 0.0,
-  "timestamp": "2026-02-01T16:00:11.852746",
+  "timestamp": "2026-02-01T16:01:18.866166",
   "vision_backbone": "repvit",
   "language_backbone": "tinyllm",
   "total_parameters": 40196257,