pytorch中如何自定义一个模型，如何建立层以及层和层之间的关系，一看就懂系列

最新推荐文章于 2024-08-02 23:48:48 发布

鼾声鼾语

最新推荐文章于 2024-08-02 23:48:48 发布

阅读量1.1k

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/pvmsmfchcs/article/details/122263555

版权

pytorch深度学习一看就懂系列和自动驾驶相关专栏收录该内容

180 篇文章 25 订阅

已下架不支持订阅

本文详细介绍了在PyTorch中如何自定义模型，包括继承nn.Module类，实现构造函数__init__定义层，以及在forward函数中构建层间关系。通过实例展示了如何在__init__中定义层的参数，如Linear层的权重和偏置，并在forward中使用torch.nn.functional进行前向运算。文章强调了forward函数的重要性，它是模型功能实现和层间连接的关键，通常实现顺序连接的层操作。

摘要由CSDN通过智能技术生成

pytorch中自定义操作都是继承nn.Module类来实现的
构造函数__init__里面，定义一系列层，forword里面实现层和层之间的关系

class Module(object):
def __init__(self):
def forward(self, *input):

def add_module(self, name, module):
def cuda(self, device=None):
def cpu(self):
def __call__(self, *input, **kwargs):
def parameters(self, recurse=True):
def named_parameters(self, prefix='', recurse=True):
def children(self):
def named_children(self):
def modules(self):  
def named_modules(self, memo=None, prefix=''):
def train(self, mode=True):
def eval(self):
def zero_grad(self):
def __repr__(self):
def __dir__(self):
'''
有一部分没有完全列出来
'''

1.定义一个类，需要首先继承nn.Module类，并且一定要实现两个基本的函数，第一是构造函数_

了解本专栏