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: