maab05 commited on
Commit
30d9ac7
·
verified ·
1 Parent(s): 05f2a72

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -4
README.md CHANGED
@@ -28,10 +28,11 @@ base_model:
28
  - resnet-18
29
  ---
30
 
31
- # Custom ResNet-18 with Integrated Dropout
32
 
33
- This is a **custom ResNet-18** model implemented in **PyTorch** with integrated **Dropout** layers.
34
- It was trained on the **CIFAR-10** dataset for image classification tasks. The model has been exported to the **ONNX format** (opset version 15).
35
- and is fully compatible with the **Aidge** platform.
 
36
 
37
  ---
 
28
  - resnet-18
29
  ---
30
 
31
+ # Custom ResNet-18 with integrated Dropout Layers
32
 
33
+ This is a **custom ResNet-18** ONNX model implemented in **PyTorch** with integrated **Dropout** layers.
34
+ It was trained on the **CIFAR-10** dataset for image classification tasks.
35
+
36
+ The model has been exported using **opset version 15** and is fully compatible with the **Aidge** platform.
37
 
38
  ---