ValueError: Expected input batch_size (324) to match target batch_size (4)

Thank You so much. It really worked. Can you please tell me how and what you checked to conclude the problem?