problem in getting same output size as batch size from forward function

[MotiBaadror]

ValueError: Expected input batch_size (128) to match target batch_size (32), it happen all the time, the expexted size is 4 times batch size. The output of forward function return probabilities 4*batchsize x num_classes. The model used is wideresnet.