首页 > 其他分享 >PyTorch构建模型训练模块

PyTorch构建模型训练模块

时间:2024-09-02 22:02:58浏览次数:10  
标签:函数 训练 模型 PyTorch 构建 模块 model 直接

前面几篇分别是数据导入模块,模型类的构建模块,损失函数模块,优化器模块,上述模块搭建好后建立训练模块,将上述几个模块在训练模块中调用完成训练。

这个不太好解释直接看模块示例:

 上面是训练模块,分别传入训练总轮次steps,实例化模型model,损失函数loss_func,但是下面训练的时候它将损失函数和反向传播优化器,更新放入了另外一个函数中,opt优化器,train_dl训练数据集,valid_dl训练数据集。

开启model.train()模式在训练完一轮后就进入model.eval()模式进行测评。

按照上面的情况将训练模块书写好后直接调用就很快。

 

还有下面这种简单粗暴的,训练模块直接不封装直接写,写了直接跑

 这个训练模块中他还增加了准确率评估计算函数如下图:

 

标签:函数,训练,模型,PyTorch,构建,模块,model,直接
From: https://www.cnblogs.com/qianzhu/p/18393636

相关文章

  • Pytorch损失函数模块
     直接调用定义就好了要使用的损失函数,但是nn.functional和nn.Module中都有损失函数,至于从哪个包中调用损失函数看你的需求,和图中所说一样,如果没有需要学习的参数就用nn.functional,如果有要学习更新的参数就用nn.Module。虽然是这样说,但是我看这个代码下面使用的就算是使用func......
  • Unity 脚本编译和构建(Build)说明(官方文档翻译校正)
    目录脚本重要概念事件函数的执行顺序脚本生命周期概述流程图的范围脚本生命周期流程图一般原则第一次场景加载(FirstSceneload)场景加载和卸载之前(Beforesceneloadandunload)Editor第一帧更新之前(Beforethefirstframeupdate)帧之间(Inbetweenframes)更新顺序(Upda......
  • PyTorch自定义模型类
    使用Pytorch自己建立一个模型类:继承初始  建立一个模型的类,一定要继承nn,Module父类,初始化:def__init__(self):super(self).__init__():如果不这样操作就使用不了继承中初始化,也使用不了继承的一些函数。 建立自己的网络层图中初始化后的self.conv都是使......
  • PyTorch导入数据
    先导入数据:将要处理的文件数据导入x_train,y_train,x_valid,y_valid中,转换为张量。 使用PyTorch中的数据模块,将上面转换为张量的数据使用TensorDataset函数转换为数据集。之后使用DataLoader来加载数据集,DataLoader包相当于一个派发器的感觉。 下方也是一个导入数据示例......
  • 有手就会之使用Dify构建RAG聊天应用(基于私有知识库和搜索引擎)
    之前我的文章里写的是通过langchain来构建RAG应用,对于很多人来说。langchain作为一个框架上手难度大,代码不够直观。但是通过dify你将学会可视化搭建工作流。什么是dify?Dify是一款开源的大语言模型(LLM)应用开发平台。它融合了后端即服务(BackendasService)和LLMOps的......
  • Python Web应用程序构建的最佳实践(代码实例与深度解析)
    在当今数字时代,构建高效、可扩展的Web应用程序是开发者们的一项重要任务。Python,作为一种简洁、强大的编程语言,为Web开发提供了丰富的工具和框架。在本篇文章中,我们将探讨使用Python构建Web应用程序的最佳实践,通过代码实例和深度解析来帮助你更好地理解和运用这些技术。1.选择合适......
  • ORM模块
    SQLAlchemy功能全面,支持的数据库多。即支持原生sql,又支持ORM操作。对异步支持,但不支持同一个session(包括asyncsession)下的并发,包括异步并发和多线程。TortoiseORM是一个完全异步的PythonORM。它是为异步操作而设计的,可以很好地与asyncio协作。允许在异步任务中并发......
  • 动手学深度学习8.1. 序列模型-笔记&练习(PyTorch)
    本节课程地址:序列模型_哔哩哔哩_bilibili本节教材地址:8.1.序列模型—动手学深度学习2.0.0documentation(d2l.ai)本节开源代码:...>d2l-zh>pytorch>chapter_multilayer-perceptrons>sequence.ipynb序列模型想象一下有人正在看网飞(Netflix,一个国外的视频网站)上的电影。......
  • 助力移动道路交通环保治理,打赢蓝天保卫战,基于YOLO家族最新端到端实时算法YOLOv10全系
    在快速发展的现代社会中,工业化的步伐虽推动了城市的繁荣,但环保问题也随之成为我们不得不面对的重大挑战。特别是在移动道路交通领域,路边与路面裸土堆积、道路扬尘等问题,不仅影响城市形象,更对空气质量与居民健康构成了严重威胁。面对这一现状,传统的监测治理手段已难以满足高效、......
  • FMC子卡设计资料:165-2路万兆光纤SFP+ FMC子卡模块
    1.概述该板卡是基于kc705和ml605的fmc10g万兆光纤扩展板设计。SFP+(10GigabitSmallFormFactorPluggable)是一种可热插拔的,独立于通信协议的光学收发器,通常传输光的波长是850nm,1310nm或1550nm,用于10Gbps的SONET/SDH,光纤通道,gigabitEthernet,10gigabitEthernet和......