2021-10-22

2021-10-22 Pytorch中创建模型的三种方法 一、单层创建 nn.Linear

代码如下（示例）：

from torch import nn
model = nn.Linear(1,1)

二、torch.nn.Sequential

按顺序进行创建

代码如下（示例）：

import torch
from torch import nn 
model = nn.Sequential(
            nn.Linear(15,1),
            nn.Sigmoid()
)

三、自定义类（继承自 nn.Module）

代码如下（示例）：

class model(nn.Module):
    def __init__(self):
        super().__init__()
        self.linear_1= nn.Linear(20,64)
        self.linear_2 = nn.Linear(64,128)
        self.linear_3 = nn.Linear(128,1)
        
    def forward(self,x):
        x = F.relu(self.linear_1(x))
        x = F.relu(self.linear_2(x))
        
        x = F.sigmoid(self.linear_3(x))
        return x

---