def __init__(self):
super(SiameseNetwork, self).__init__()
#self.resnet = models.resnet152(pretrained=True)
#self.resnet = models.resnet101(pretrained=True)
self.resnet = mod_res.resnet101(filter_size=3)
def __init__(self):
super(SiameseNetwork, self).__init__()
self.resnet = mod_res.resnet101(filter_size=3)
if enableEval:
self.resnet.load_state_dict(
torch.load('./pretrainedWeights/resnet101_lpf3.pth.tar')
['state_dict'])