Brain Tumor Segmentation with Shuffle Attention

Hello everyone…

I want to train this modified3dunet with input seize 128x128x128, but i got error
RuntimeError: The size of tensor a (18) must match the size of tensor b (256) at non-singleton dimension 2

Could anyone help me please?
This is my model architecture