def __init__(self): ut.super2(Siamese, self).__init__() self.resnet = torchvision.models.resnet50(pretrained=True) self.num_fcin = self.resnet.fc.in_features # replace the last layer of resnet self.resnet.fc = torch.nn.Linear(self.num_fcin, 500) self.pdist = torch.nn.PairwiseDistance(p=2)
def __init__(self, margin=1.0): ut.super2(ContrastiveLoss, self).__init__() self.margin = margin