AIGC原理与实践:零基础学大语言模型、扩散模型和多模态模型
上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人

2.1.3 构建模型

使用PyTorch实现神经网络的关键就是选择网络层,构建模型,然后选择损失函数和优化器。在nn工具箱中,可以直接引用的网络很多,有全连接层、卷积层、循环层、归一化层、激活层等。

PyTorch构建模型大致有以下两种方式。

1)继承nn.Module基类来构建模型。

2)使用PyTorch Lightning来构建模型。