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

Thanks. It really helps.