ONNX 模型库
返回模型

说明文档

CustomCNN (ONNX)

这是一个在 CIFAR-10 数据集上训练的自定义卷积神经网络 (CNN),用于测试 Aidge 平台的自定义 Dropout 算子集成。

详情

  • 架构:带有 Dropout 层的自定义卷积神经网络 (CNN)
  • 训练数据集:CIFAR-10(60,000 张 32x32 彩色图像,10 个类别)
  • 数据归一化mean = [0.4914, 0.4822, 0.4465] ; std = [0.2023, 0.1994, 0.2010]
  • Dropout 概率:0.3
  • ONNX opset 版本:15
  • 转换工具:PyTorch → ONNX

EclipseAidge/CustomCNN

作者 EclipseAidge

image-classification
↓ 0 ♥ 0

创建时间: 2025-06-13 09:00:41+00:00

更新时间: 2026-03-17 14:55:18+00:00

在 Hugging Face 上查看

文件 (5)

.gitattributes
CustomCNN.onnx ONNX
CustomCNN_cleaned.onnx ONNX
MarwaNet.onnx ONNX
README.md