def init_pretrained(self, pretrained: str=None) -> None:
    if pretrained:
        self.context_path.backbone.load_state_dict(torch.load(pretrained, map_location='cpu'), strict=False)
def init_pretrained(self, pretrained: str=None) -> None:
    if pretrained:
        self.context_path.backbone.load_state_dict(torch.load(pretrained, map_location='cpu'), strict=False)