ValueError: Expected input batch_size (20) to match target batch_size (8).

[zhanjunyao]

after I prepared data and trained base model as Readme，the error happended in region_loss.py
loss_cls = self.class_scale * ClassificationLoss(cls, tcls)
I don't know where goes error,someone has ever met this problem?how to solve?
thank you