实验步骤
搭建resnet18网络
数据集加载
模型训练和改进
分析评估
Kaggle提交
网络构建
实验初期拟采用torchvision中实现的resnet18作为网络结构,为了方便修改网络结构,于是重新实现了resnet18网络
resnet18由一个7×7的降采样卷积,一个max pooling层,8个basicblock,一个全局池化层,最后接一个全连接层组成,如下图
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/
转载文章受原作者版权保护。转载请注明原作者出处!