当前位置: 首页 > 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

相关文章:

  • 物联网(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删除
  • 建行原副行长章更生涉嫌受贿罪、违法发放贷款罪被逮捕
  • 财政部:鼓励政策性银行对符合条件的城市更新项目提供支持
  • 减重人生|吃得越少越好?比体重秤上的数字,更有意义的是什么?
  • 俄乌上周在土耳其直接谈判,外交部回应
  • 一周人物|收藏家瓦尔特捐出藏品,女性艺术家“对话”摄影
  • 河南发布高温橙警:郑州、洛阳等地最高气温将达40℃以上