Extension to different input size

[JenniferVandoni]

I'm facing problems setting a different input size (e.g. 32,160,160 instead of 16,160,160) because dimensions mismatch (I'm getting ValueError: Expected input batch_size (128) to match target batch_size (64)). The problem is in x = x.view(-1,sizes[1],sizes[3],sizes[4]) and I think it should be x = x.view(-1,sizes[1]*sizes[2],sizes[3],sizes[4]), but how to change the layer accordingly?
Thank you