def __init__(self, in_channels, num_classes):
    super().__init__()
    self.cls_score = nn.Linear(in_channels, num_classes)
    self.bbox_pred = nn.Linear(in_channels, (num_classes * 4))
def __init__(self, in_channels, num_classes):
    super().__init__()
    self.cls_score = nn.Linear(in_channels, num_classes)
    self.bbox_pred = nn.Linear(in_channels, (num_classes * 4))