当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch Module用法及代码示例


本文简要介绍python语言中 torch.nn.Module 的用法。

用法:

class torch.nn.Module

变量

training(bool) -布尔值表示该模块是处于训练模式还是评估模式。

所有神经网络模块的基类。

你的模型也应该继承这个类。

模块还可以包含其他模块,允许将它们嵌套在树结构中。您可以将子模块分配为常规属性:

import torch.nn as nn
import torch.nn.functional as F

class Model(nn.Module):
    def __init__(self):
        super(Model, self).__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))

以这种方式分配的子模块将被注册,并且当您调用 to() 等时,它们的参数也会被转换。

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.Module。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。