2.25 深度学习模型构建全解:从nn.Module到RetNet18

继承nn.Module基类并应用模型容器构建模型:

1.使用nn.Sequential模型容器

2.使用nn.ModuleList模型容器

3.使用nn.ModuleDict模型容器

 

自定义网络模块:·残差块有两种,一种是正常的模块方式,将输入与输出相加,然后应用激活函数ReLU。

另一种是为使输入与输出形状一致,需添加通过1×1卷积调整通道和分辨率。

·组合这两个模块得到现代经典RetNet18网络结构。

训练模型:

1.加载预处理数据集

2.定义损失函数

3.定义优化方法

4.循环训练模型

5.循环测试或验证模型

6.可视化结果

你可能感兴趣的:(机器学习,算法,人工智能)