resnet18实现cifar10分类

实验步骤

搭建resnet18网络

数据集加载

模型训练和改进

分析评估

Kaggle提交

网络构建

实验初期拟采用torchvision中实现的resnet18作为网络结构,为了方便修改网络结构,于是重新实现了resnet18网络

resnet18由一个7×7的降采样卷积,一个max pooling层,8个basicblock,一个全局池化层,最后接一个全连接层组成,如下图

resnet18实现cifar10分类resnet18实现cifar10分类

tensorboard网络结构可视化,左图为torchvision中的resnet实现,右图为自定义实现

代码如下

; 定义残差块

class IdentityBlock(nn.Module

Original: https://blog.csdn.net/qq_41335232/article/details/122232993
Author: 孤独腹地
Title: resnet18实现cifar10分类

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/688249/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球