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