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

SegFormer之模型训练

单卡训练,所有配置文件里的【SyncBN】改为【BN】

启动训练

(1)终端直接运行
python tools/train.py local_configs/segformer/B1/segformer.b1.512x512.ade.160k.py
(2)在编辑器中运行
在这里插入图片描述
在 [config] 前面加上’–‘将 [config] 由必填参数改为可选参数,并赋予默认值default=‘…’,然后就可以在编辑器中run或者debug了

关于配置文件

以**/local_configs/segformer/B0/segformer.b0.512x512.ade.160k.py**为例
在这里插入图片描述
mmcv下有一个核心组件Config,它提供各种格式的配置文件解析功能,包括 py、json、ymal 和 yml。
在这里插入图片描述
Config.fromfile(args.config)可以解析segformer.b0.512x512.ade.160k.py和内部的_base_中指定的其余配置文件,并将他们合并成一个配置内容,返回一个Config对象cfg。合并的规则是:segformer.b0.512x512.ade.160k.py中的配置内容会替换掉其余配置文件中相同key下的内容

【segfomer.py】
在这里插入图片描述
主要是模型的相关设置

【ade20k.py】
在这里插入图片描述
主要是训练数据的相关设置,注意这里的[data_root]是训练数据所在的路径

【default_runtime.py】
在这里插入图片描述
这部分暂时不太清除具体的作用,似乎和日志有关

【schedule_160k_adamw.py】
在这里插入图片描述
优化器、学习率和迭代次数设置

http://www.lryc.cn/news/125053.html

相关文章:

  • Azure资源命名和标记决策指南
  • 【在一个升序数组中插入一个数仍升序输出】
  • 图像去雨、去雪、去雾论文学习记录
  • YARN框架和其工作原理流程介绍
  • 多维时序 | MATLAB实现ZOA-CNN-BiGRU-Attention多变量时间序列预测
  • centos上下载redis
  • 黑马项目一阶段面试58题 Web14题(二)
  • 软考高项-思维导图34-36(计算机高级系统项目管理师)
  • C++的stack和queue+优先队列
  • Ubuntu 18.04.6 Android Studio Giraffe adb logcat 无法使用
  • Python采集天气数据,做可视化分析【附源码】
  • 优维低代码实践:自定义模板
  • 电商3D产品渲染简明教程
  • 探索未来:元宇宙与Web3的无限可能
  • GraphQL(六)登录态校验Directive
  • 强大的AI语言模型
  • 成集云 | 鼎捷ERP采购单同步钉钉 | 解决方案
  • 【Kubernetes】Kubernetes的PV和PVC的用法
  • 【Redis】Redis三种集群模式-主从、哨兵、集群各自架构的优点和缺点对比
  • Python爬虫:如何使用Python爬取网站数据
  • 剑指offer专题2:队列和栈
  • pytorch入门-神经网络
  • kafka使用心得(二)
  • (二)掌握最基本的Linux服务器用法——Linux下简单的C/C++ 程序、项目编译
  • 应急响应-钓鱼邮件的处理思路溯源及其反制
  • Hadoop Hbase Hive 版本对照一览
  • Postgresql 基础使用语法
  • Qt 之 QDebug,QString
  • 【C++】面试题
  • 机器学习算法之-逻辑回归(1)