maab05 commited on
Commit
6ee4dff
·
verified ·
1 Parent(s): 6a99d4e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -39,7 +39,8 @@ The model has been exported using **opset version 15** and is fully compatible w
39
  - **Architecture**: Customized ResNet-18 with integrated Dropout layers
40
  - **Trained on**: CIFAR-10 (60,000 32x32 color images, 10 classes)
41
  - **Image Preprocessing**: Images were resized to `128×128`
42
- - **Data Normalization**: `mean = [0.4914, 0.4822, 0.4465]`, `std = [0.2023, 0.1994, 0.2010]`
 
43
  - **Dropout Probability**: 0.3
44
  - **ONNX opset version**: 15
45
  - **Conversion tool**: PyTorch → ONNX
 
39
  - **Architecture**: Customized ResNet-18 with integrated Dropout layers
40
  - **Trained on**: CIFAR-10 (60,000 32x32 color images, 10 classes)
41
  - **Image Preprocessing**: Images were resized to `128×128`
42
+ - **Data Normalization**: `mean = [0.4914, 0.4822, 0.4465]` ;
43
+ `std = [0.2023, 0.1994, 0.2010]`
44
  - **Dropout Probability**: 0.3
45
  - **ONNX opset version**: 15
46
  - **Conversion tool**: PyTorch → ONNX