PyTorch利用Torch.nn构建卷积神经网络
一、概要
1.需要声明的是构建卷积神经网络需要有一定的面向对象基础,因为所有建立的模型结构都是继承自nn.Module这个基类所完成的,面向对象的基础知识可以参考我之前的博客
面向对象基础 python
2.我们需要新建一个子类,并且构造函数和前向传播等方法需要被重写才能实现自己编写的网络
前向传播和反向传播原理 python实现
3.我们还需要知道torch中卷积层,池化层,全连接层等部分的编写方法和拼接方式
1.需要声明的是构建卷积神经网络需要有一定的面向对象基础,因为所有建立的模型结构都是继承自nn.Module这个基类所完成的,面向对象的基础知识可以参考我之前的博客
面向对象基础 python
2.我们需要新建一个子类,并且构造函数和前向传播等方法需要被重写才能实现自己编写的网络
前向传播和反向传播原理 python实现
3.我们还需要知道torch中卷积层,池化层,全连接层等部分的编写方法和拼接方式