集成 Dropout 层的自定义 ResNet-18

这是一个在 PyTorch 中实现的 自定义 ResNet-18 ONNX 模型，集成了 Dropout 层。 该模型在 CIFAR-10 数据集上训练，用于图像分类任务。 该模型使用 opset 版本 15 导出，完全兼容 Aidge 平台。

详情

• 架构：集成 Dropout 层的自定义 ResNet-18
• 训练数据集：CIFAR-10（60,000 张 32×32 彩色图像，10 个类别）
• 图像预处理：图像调整为 128×128
• 数据标准化：mean = [0.4914, 0.4822, 0.4465] ; std = [0.2023, 0.1994, 0.2010]
• Dropout 概率：0.3
• ONNX opset 版本：15
• 转换工具：PyTorch → ONNX