Capo_Mestre:

out = self.bn(out)


It means, out in this line has a shape that is different from [64, 4, 4] that is why you getting size mismatch error.
Try to put a print(out) right after the aforementioned line and then use the sizes for out.view(-1, <printed_size>).