search

NVlabs / SegFormer

Official PyTorch implementation of SegFormer
https://arxiv.org/abs/2105.15203
Other
2.58k stars 357 forks source link

single-gpu training is ok, but Multi-gpu training has some wrong result #144

Open zhedahe opened 10 months ago

zhedahe commented 10 months ago

hi, everyone. my single-gpu training is ok, but Multi-gpu training is not successful in evaluaton period. its wrong info: IndexError: boolean index did not match indexed array along dimension 0; dimension is XXX but corresponding boolean dimension is YYY XXX and YYY is a test image size I check my dataset, it is ok, and can be used in other training model, such as deeplab. so anybody can help me? thanks a lot!