maab05 commited on
Commit
5e641c6
·
verified ·
1 Parent(s): 476b22e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -2
README.md CHANGED
@@ -1,8 +1,38 @@
1
- **MarwaNet** is a custom convolutional neural network (CNN) trained on the CIFAR-10 dataset, developed to test the integration of a custom Dropout operator for the Aidge platform.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
  ## Model Details
4
 
5
  - **Architecture**: Custom Convolutional Neural Network (CNN)
6
  - **Trained on**: CIFAR-10 (60,000 32x32 color images in 10 classes)
7
- - **ONNX opset version**: 11
8
  - **Conversion tool**: PyTorch → ONNX
 
1
+ ---
2
+ license: cc
3
+ tags:
4
+ - onnx
5
+ - image-classification
6
+ - cifar10
7
+ - dropout
8
+ - aidge
9
+ pipeline_tag: image-classification
10
+ datasets:
11
+ - cifar10
12
+ metrics:
13
+ - type: accuracy
14
+ value: 69.96%
15
+ model-index:
16
+ - name: Custom ResNet-18 with Integrated Dropout
17
+ results:
18
+ - task:
19
+ type: image-classification
20
+ name: Image Classification
21
+ dataset:
22
+ name: CIFAR-10
23
+ type: cifar10
24
+ metrics:
25
+ - type: accuracy
26
+ value: 69.96%
27
+ ---
28
+
29
+ # MarwaNet (ONNX)
30
+
31
+ This is a **custom convolutional neural network (CNN)** trained on the **CIFAR-10** dataset, developed to test the integration of a custom Dropout operator for the Aidge platform.
32
 
33
  ## Model Details
34
 
35
  - **Architecture**: Custom Convolutional Neural Network (CNN)
36
  - **Trained on**: CIFAR-10 (60,000 32x32 color images in 10 classes)
37
+ - **ONNX opset version**: 15
38
  - **Conversion tool**: PyTorch → ONNX