Update README.md

README.md

@@ -32,7 +32,12 @@ This is a **custom convolutional neural network (CNN)** trained on the **CIFAR-1
 
 ## Details
 
-- **Architecture**: Custom Convolutional Neural Network (CNN)
-- **Trained on**: CIFAR-10 (60,000 32x32 color images in 10 classes)
+- **Architecture**: Custom Convolutional Neural Network (CNN) with a Dropout layer
+- **Trained on**: CIFAR-10 (60,000 32x32 color images, 10 classes)
+- **Data Normalization**: `mean = [0.4914, 0.4822, 0.4465]` ; `std = [0.2023, 0.1994, 0.2010]`
+- **Dropout Probability**: 0.3
 - **ONNX opset version**: 15
-- **Conversion tool**: PyTorch → ONNX
+- **Conversion tool**: PyTorch → ONNX
+
+
+