search

bowenc0221 / panoptic-deeplab

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)
Apache License 2.0
590 stars 117 forks source link

RuntimeError: each element in list of batch should be of equal size #62

Closed CuberrChen closed 3 years ago

CuberrChen commented 3 years ago

hi！i'm trying to understand your code。but when i run " python -m torch.distributed.launch --nproc_per_node=2 train_net.py --cfg ../configs/panoptic_deeplab_MBNV2_100_os32_cityscapes_bs16_lr_x2.yaml " as your README,the error happend! can you help me? i didn't do anything to the code.

bowenc0221 commented 3 years ago

Which PyTorch version are you using? Can you try 1.5.0 as mentioned here: https://github.com/pytorch/pytorch/issues/42654

CuberrChen commented 3 years ago

thanks! it is the reason of Pytoch version!