当前位置: 首页 > 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.xdnf.cn/news/543925.html

相关文章:

  • Docker run -v 的 rw 和 ro 模式_docker ro
  • React Flow 数据持久化:Django 后端存储与加载的最佳实践(含详细代码解析)
  • React中使用ahooks处理业务场景
  • 网络攻防技术
  • 微软押注“代理式AI网络”:一场重塑软件开发与工作方式的技术革命
  • 网络攻击通常会被分为哪些类型?
  • FreeSWITCH rtcp-mux 测试
  • Hbuilder X4.65新建vue3项目存在的问题以及解决办法
  • HTB 赛季8靶场 - Puppy
  • 对未来软件的看法
  • Vue响应式系统演进与实现解析
  • Linux系统之traceroute命令详解:追踪网络路径的核心工具
  • Linux网络编程:广播、组播与原始套接字
  • 51单片机编程学习笔记——无源蜂鸣器演奏《祝你生日快乐》
  • 计算机网络通信技术与协议(七)———关于ACL的详细解释
  • 高密度服务器机柜散热方案:高风压风机在复杂风道中的关键作用与选型要点
  • 电力设备智能化方案复盘
  • DataLight(V1.7.12)版本更新发布
  • 通义灵码助力Neo4J开发:快速上手与智能编码技巧
  • 钉钉开发之AI消息和卡片交互开发文档收集
  • React的合成事件(SyntheticEventt)
  • 企业终端设备的安全管控
  • 【Tauri2】046—— tauri_plugin_clipboard_manager(一)
  • RK3588 ArmNN CPU/GPU ResNet50 FP32/FP16/INT8 推理测试
  • Qt项目开发中所遇
  • 《Android 应用开发基础教程》——第十三章:权限管理机制与运行时权限请求(以拍照/存储为例)
  • MySQL——基本查询内置函数
  • 从零开始的抽奖系统创作(2)
  • vue3/vue2大屏适配
  • 基于统计检验与多模型对心脏病数据的分析与预测