【深度学习】Pytorch学习笔记(一)

Pytorch中需要自己定义网络模型,该模型需要封装到一个自定义的类中,该类只是一个子类,其继承的父类为torch.nn.Module,可见如下树形结构图: Module实际又是继承了Object类,关于为什么要继承Object类有兴趣的可以看这篇博客MRO算法 也就是说,自定义的模型除了要有__init__()构造函数之外,还必须有forward函数,重写父类Module的该方法。 在子类的__
相关文章
相关标签/搜索