Number of class in training

[mfanparc]

Hi, thanks for the inspiring work. When I retrained the model with the config /oln_box, i received this: "AssertionError: The num_classes (1) in Shared2FCBBoxScoreHead of MMDataParallel does not matches the length of CLASSES 80) in CocoSplitDataset " So, should I set the num_classes = 80 instead of 1 when training with COCO? Or there is something else I should do. Thanks!