2022年3月16日
Pytorch03——模型定义
摘要： 模型搭建基础知识： 1.Module类是torch.nn模块里提供的一个模型构造类(nn.Module)，是所有神经网络模块的基类，在实现神经网络的时候是通过继承它来定义我们想要的模型。 2.Pytorch模型定义主要是两部分一部分是 各部分的初始化（init）；数据流向定义（forward），基于 阅读全文
posted @ 2022-03-16 14:59 TCcjx 阅读(50) 评论(0) 推荐(0) 编辑