predict mask only 1/4 of original image

[sparkfax]

image size(3,512,512)
mask_pred = self.mask_classifier(out_ss)
print(mask_pred.shape) #torch.Size([16, 5, 128, 128])
target mask 512, predict mask ony 128

self.gau_block1 = GAU(channels_blocks[0], channels_blocks[1], upsample=False)
self.gau_block2 = GAU(channels_blocks[1], channels_blocks[2])
self.gau_block3 = GAU(channels_blocks[2], channels_blocks[3])

torch.Size([16, 2048, 32, 32])
False
True
torch.Size([16, 512, 64, 64])
True
torch.Size([16, 256, 128, 128])
torch.Size([16, 5, 128, 128])

/opt/conda/lib/python3.6/site-packages/catalyst/dl/utils/criterion/dice.py in dice(outputs, targets, eps, threshold, activation)
30 outputs = (outputs > threshold).float()
31
---> 32 intersection = torch.sum(targets * outputs)
33 union = torch.sum(targets) + torch.sum(outputs)
34 dice = 2 * intersection / (union + eps)

RuntimeError: The size of tensor a (512) must match the size of tensor b (128) at non-singleton dimension 3