SegFormer之模型训练_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】

优化器、学习率和迭代次数设置