DEV Community
·
PyTorch中的模块
💡
原文英文，约400词，阅读约需2分钟。
📝
内容提要
本文讲解如何用PyTorch创建自定义模型类`MyModel`，包含多个线性层和ReLU激活函数。通过`state_dict()`获取模型状态，`parameters()`返回参数迭代器。`train()`和`eval()`用于切换训练和评估模式。示例代码展示了模型参数定义、前向传播及模式切换。
🎯
关键要点
- 使用PyTorch创建自定义模型类MyModel，继承自nn.Module。
- 模型包含多个线性层和ReLU激活函数。
- state_dict()方法返回模型的状态字典。
- parameters()方法返回模型参数的迭代器。
- num3和num4未使用Parameter()定义，因此不在state_dict()和parameters()中显示。
- train()方法用于设置模型为训练模式。
- eval()方法用于设置模型为评估模式。
- 示例代码展示了模型参数的定义和前向传播过程。
- 通过torch.manual_seed(42)设置随机种子以确保可重复性。
➡️
