Update README.md
Browse files
README.md
CHANGED
|
@@ -36,4 +36,11 @@ It was trained on the **CIFAR-10** dataset for image classification tasks.
|
|
| 36 |
|
| 37 |
The model has been exported using **opset version 15** and is fully compatible with the **Aidge** platform.
|
| 38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
---
|
|
|
|
| 36 |
|
| 37 |
The model has been exported using **opset version 15** and is fully compatible with the **Aidge** platform.
|
| 38 |
|
| 39 |
+
## Details
|
| 40 |
+
|
| 41 |
+
- **Architecture**: ResNet-18 (customized)
|
| 42 |
+
- **Trained on**: CIFAR-10 (60,000 32x32 color images in 10 classes)
|
| 43 |
+
- **ONNX opset version**: 15
|
| 44 |
+
- **Conversion tool**: PyTorch → ONNX
|
| 45 |
+
|
| 46 |
---
|