单卡训练,所有配置文件里的【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】
优化器、学习率和迭代次数设置