top of page
FOUND IN
C:\Users\Gebruiker\AppData\Roaming\Visions of Chaos\Examples\MachineLearning\HeadTrip\dream.py
def init_model(self):
"""initializes the model with the config file"""
if self.config["model"] == "resnet":
network = models.resnext50_32x4d(pretrained=True)
elif self.config["model"] == "vgg19":
network = models.vgg19(pretrained=True)
elif self.config["model"] == "densenet":
network = models.densenet121(pretrained=True)
elif self.config["model"] == "inception":
network = models.inception_v3(pretrained=True)
elif self.config["model"] == "mobile":
network = models.mobilenet_v2(pretrained=True)
elif self.config["model"] == "shuffle":
network = models.shufflenet_v2_x0_5(pretrained=True)
elif self.config["model"] == "squeeze":
network = models.squeezenet1_1(pretrained=True)
elif self.config["model"] == "resnetx":
network = models.resnext101_32x8d(pretrained=True)
elif self.config["model"] == "masnet":
network = models.mnasnet1_0(pretrained=True)
elif self.config["model"] == "googlenet":
network = models.googlenet(pretrained=True)
elif self.config["model"] == "alexnet":
network = models.alexnet(pretrained=True)
else:
bottom of page