multi gpu error

[cswwp]

It's a strange error when i inference model efficientb5 unet with tensor with shape 43128800, seems torch.cat([x, blocks.popitem()[1]], dim=1) error, i print shape as upper image, but it shows only in some batchs, i'm sure input tensor shape is all same (batch3128800) anyone help?

[zhoudaxia233]

@NPU-Franklin Franklin created a PR (#11 ) to support multi GPUs. I do not have multi cards therefore I cannot test it. But maybe you can give it a try.