mmseg 单机多卡训练设置

bash tools/dist_train.sh "配置文件" "GPU数量"

示例:

bash tools/dist_train.sh config.py 2

同时在配置文件中

train_dataloader = dict(
    batch_size=4,
    ...    
}

这里的batch_size是4指的是每张卡上的batch_size而不是总的batch_size,总的batch_size是卡的数量乘单张卡的batch_size。
2*4=8

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐