about the gpu memory consumed

[chenchr]

Hello. Thanks for the code.
In the paper, during inference, there will be two tensor with size b * (2*h*w) * h * w in the psa module, willl it consumes too much memory ?

[hszhao]

Hi, @chenchr, for input tensors with larger spatial size like cityscapes data, a shrinking/downsampling operation will be done before fed into PSA module.