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

一、神经网络基本框架

1.查看官方文档

• 第一步
  

• 第二步
  

• 第三步：不同方法的作用
  

• 第四步：containers的讲解
  

自己创建模型的话：重写init和forward

forward函数：x是模型输入，conv是卷积，relu是非线性处理

二、案例

1.代码

from torch import nn
import torch
class Tudui(nn.Module):
    def __init__(self) :
        super().__init__()

    def forward(self,input):
        output=input+1
        return output

tudui=Tudui() #创建网络
x=torch.tensor(1.0) #定义输出的参数
output=tudui(x) #将参数输入到网络
print(output)   #输出参数

重写方法：在类中点击菜单栏中的code->Generate->overrite method

2.debug了解流程

注意debug：1.断点 2.选择在自己的代码中执行