扬帆远航_df7c

关注

6.nn.Module的使用

扬帆远航_df7c

关注

阅读 18

2022-04-15

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

    • 1.

  • 模板
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.

相关推荐

大南瓜鸭

【05】nn.Module模块使用

大南瓜鸭 69 0 0

陆佃

nn.Module类的讲解

陆佃 55 0 0

半夜放水

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

半夜放水 49 0 0

流计算Alink

pytorch报错 No module named 'nn'

流计算Alink 80 0 0

清冷的蓝天天

Pytorch:torch.nn.Module

清冷的蓝天天 22 0 0

小飞侠熙熙

pytorch中nn.functional和nn.Module的区别

小飞侠熙熙 41 0 0

Python百事通

关于Class继承nn.Module和nn.Sequential的区别

Python百事通 74 0 0

梦为马

神经网络-nn.module

梦为马 73 0 0

十里一走马

Pytorch-----torch.nn.Module.modules()

十里一走马 56 0 0

少_游

搭建深度学习框架+nn.Module

少_游 25 0 0

精彩评论(0)

0 0 举报