当前位置: 首页 > news >正文

深度学习-runner.run(data_loaders, cfg.workflow)内部执行过程

文件:~/catkin_ws/SparseDrive/projects/mmdet3d_plugin/apis/mmdet_train.py
完成数据加载器、优化器、运行器实例化后,

@RUNNERS.register_module()
class IterBasedRunner(BaseRunner):"""Iteration-based Runner.This runner train models iteration by iteration."""def train(self, data_loader, **kwargs):#若self.model是torch.nn.Module的子类且未重写train()方法​​:#会调用PyTorch原生的torch.nn.Module.train()方法self.model.train()self.mode = 'train'self.data_loader = data_loaderself._epoch = data_loader.epochdata_batch = next(data_loader)self.data_batch = data_batchself.call_hook('before_train_iter')outputs = self.model.train_step(data_batch, self.optimizer, **kwargs)if not isinstance(outputs, dict):raise TypeError('model.train_step() must return a dict')if 'log_vars' in outputs:self.log_buffer.update(outputs['log_vars'], outputs['num_samples'])self.outputs = outputsself.call_hook('after_train_iter')del self.data_batchself._inner_iter += 1self._iter += 1
http://www.dtcms.com/a/201985.html

相关文章:

  • 物联网(IoT)智能项目全景指南:技术构架、实现细节与应用实践
  • 以太联 - Intellinet 闪耀台北 SecuTech 国际安全科技应用博览会
  • 【强化学习】深度强化学习 - Deep Q-Network(DQN)算法
  • JAVA虚拟机有义务保证<clinit>()方法的线程安全
  • onlyoffice 源码 调试说明 -ARM和x86双模式安装支持
  • # YOLOv5:目标检测的新里程碑
  • 【计算机方向海外优质会议推荐】第二届图像处理、机器学习与模式识别国际学术会议(IPMLP 2025)
  • OpenHarmony 5.0设置应用设置手势导航开关打开后重新关闭导航栏和设置界面重合
  • 【漫话机器学习系列】269.K-Means聚类算法(K-Means Clustering)
  • 【C++】vector模拟实现
  • 优化model我们可能会怎么做(具体)
  • 正则表达式进阶(三):递归模式与条件匹配的艺术
  • HarmonyOS5云服务技术分享--云存储SDK文章整理
  • 数据库基础
  • 【[特殊字符] Vue 3 实现动态加载子组件并缓存状态完整指南】
  • uniapp生成的app,关于跟其他设备通信的支持和限制
  • 48、c# 中 IList 接⼝与List的区别是什么?
  • 深入解析Spring Boot与Redis的缓存集成实践
  • Spark Core基础与源码剖析全景手册
  • Java转Go日记(四十一):Gorm删除
  • 【iOS】类结构分析
  • 中间件-seata
  • [Linux] Linux线程信号的原理与应用
  • (二十四)Java网络编程全面解析:从基础到实践
  • 在 Excel 中使用通义灵码辅助开发 VBA 程序
  • LeetCode 1345. 跳跃游戏 IV(困难)
  • ZooKeeper 原理解析及优劣比较
  • Gartner《AI Infrastructure WithKubernetes参考架构》学习心得
  • LabVIEW下AI开发
  • 杰里7006d日志分析