0
点赞
收藏
分享

微信扫一扫

6.nn.Module的使用

扬帆远航_df7c 2022-04-15 阅读 18
python

神经网络的基本骨架-nn.Module的使用

  • 模板
import torch.nn as nn
import torch.nn.functional as F

class Model(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(1, 20, 5)
        self.conv2 = nn.Conv2d(20, 20, 5)

    def forward(self, x):
        x = F.relu(self.conv1(x))
        return F.relu(self.conv2(x))

1.

举报

相关推荐

0 条评论