Pytorch系列之——模型创建与nn.Module

模型创建与nn.Module 网络模型创建步骤 nn.Module属性 在学习模型创建步骤之前,回顾一下前面提到的机器学习模型训练的五大步骤,即:数据、模型、损失函数、优化器和迭代训练。这次我们主要谈论模型模块这部分。 对于模型模块,首先分为两大部分分别是模型的创建和模型权值的初始化。其中在模型创建模块,有分为构建网络层和拼接网络层这两个子模块,我们通过构建类似卷积层、池化层和激活函数层等子模块,
相关文章
相关标签/搜索