Update README.md

README.md

@@ -66,12 +66,23 @@ model.eval()
 
 print("✅ KiyEngine V3 ready for inference.")
 
-# Example forward pass (replace with your actual input)
-# batch_size = 1, sequence_length = 64 (chess board representation)
+# Create a fake input(Batch=1, Seq_len=64)
 dummy_input = torch.randint(0, 768, (1, 64))
+
 with torch.no_grad():
+    # Gọi model
     output = model(dummy_input)
-    print(f"Output shape: {output.shape}")
+
+    # ✅ Cách lấy dữ liệu chuẩn từ KiyEngineOutput
+    print(f"🎉 Success!")
+    print(f"1. Policy Logits (Dự đoán nước đi): {output.policy_logits.shape}") 
+    # Kỳ vọng: torch.Size([1, 768]) -> Dự đoán xác suất cho 768 nước đi có thể
+    
+    print(f"2. Value (Đánh giá thế cờ):      {output.value.shape}")
+    # Kỳ vọng: torch.Size([1, 1])   -> Điểm số từ -1 (Thua) đến 1 (Thắng)
+    
+    print(f"3. Last Hidden State (Tư duy):   {output.last_hidden_state.shape}")
+    # Kỳ vọng: torch.Size([1, 64, 384])
 ```
 
 ---