search

rstrudel / segmenter

[ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation
MIT License
846 stars 175 forks source link

Mutli-GPUs training #29

Closed qiulesun closed 3 years ago

qiulesun commented 3 years ago

This is a good paper and very interested idea! There is a training cmd using a single gpu in readme. For multi-gpus training, could you provide the corresponding cmd ?

rstrudel commented 3 years ago

Hi @qiulesun , Please check https://github.com/rstrudel/segmenter/issues/22#issuecomment-918863351 , I provide here a slurm batch script to do multi-gpu training.

qiulesun commented 3 years ago

The mIoU value of Seg-L-Mask/16 on ADE20K using resolution of 512x512 is not listed in readme (only contain that trained using resolution 640x640). In terms of this evaluation, do you consider providing such value?

rstrudel commented 3 years ago

Sure, on ADE20K for a resolution of 512x512, the mIoU for single-scale/multi-scale is 51.3 / 53.2 .

rstrudel commented 3 years ago

I am closing this issue, feel frre to reopen it if you have more questions.