Ejemplos de resnet101 en Python

Ejemplo n.º 1

Archivo: evalAll.py Proyecto: y1337x/yugioh-one-shot-learning

    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)

Ejemplo n.º 2

Archivo: dataLoaderEval.py Proyecto: y1337x/yugioh-one-shot-learning

    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'])