def __init__(self, num_polars, num_channels, num_classes):
    super(PolarInst, self).__init__()
    self.num_classes = num_classes
    self.backbone = StageBackbone()
    self.fpn = FeaturePyramidNet(num_channels)
    self.polar_head = PolarHead(num_polars, num_channels, num_classes)
    self.distance_scales = [nn.Parameter(torch.tensor(1.0, dtype=torch.float)) for _ in range(5)]
def __init__(self, num_polars, num_channels, num_classes):
    super(PolarInst, self).__init__()
    self.num_classes = num_classes
    self.backbone = StageBackbone()
    self.fpn = FeaturePyramidNet(num_channels)
    self.polar_head = PolarHead(num_polars, num_channels, num_classes)
    self.distance_scales = [nn.Parameter(torch.tensor(1.0, dtype=torch.float)) for _ in range(5)]