每天讲解一点PyTorch 14模型定义,继承nn.Module

Posted cv.exp

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了每天讲解一点PyTorch 14模型定义,继承nn.Module相关的知识,希望对你有一定的参考价值。

今天我们讲解class的定义和实现:

class Dec(nn.Module):#继承nn.Module类
    def __init__(self, ***, ***):#构建模块
        super(Dec, self).__init__()#父类初始化
        self.***= ***
        self.***= ***
        
    def forward(self, x, ***):#组装
        x = ***
        return self.***(x)
        
    def initialize_weights(self):#权值初始化
        ***
net = Dec()#实例化
outputs = net(inputs)

torch.nn.Module
torch.nn

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))

下一节我们讲怎么使用nn.Sequential定义模型

以上是关于每天讲解一点PyTorch 14模型定义,继承nn.Module的主要内容,如果未能解决你的问题,请参考以下文章

每天讲解一点PyTorch nn.Embedding

每天讲解一点PyTorch nn.Embedding

每天讲解一点PyTorch 18多卡训练torch.nn.DataParallel

每天讲解一点PyTorch 18多卡训练torch.nn.DataParallel

每天讲解一点PyTorch 18多卡训练torch.nn.DataParallel

每天讲解一点PyTorch F.softmax