Python get_model Examples

Example #1

File: main.py Project: liuyixin-louis/DCP_py3

    def _set_model(self):
        """
        Available model
        cifar:
            preresnet
        imagenet:
            resnet
        """

        self.original_model, self.test_input = get_model(
            self.settings.dataset, self.settings.net_type, self.settings.depth,
            self.settings.n_classes)
        self.pruned_model, self.test_input = get_model(self.settings.dataset,
                                                       self.settings.net_type,
                                                       self.settings.depth,
                                                       self.settings.n_classes)

Example #2

    def _set_model(self):
        """
        Get model
        """

        self.pruned_model, self.test_input = get_model(self.settings.dataset,
                                                       self.settings.net_type,
                                                       self.settings.depth,
                                                       self.settings.n_classes)
        self.replace_layer_with_mask_conv()

Example #3

File: main.py Project: qdmy/DCP-1

    def _set_model(self):
        """
        Available model
        cifar:
            preresnet
        imagenet:
            resnet
        """

        self.model, _ = get_model(self.settings.dataset,
                                  self.settings.net_type, self.settings.depth,
                                  self.settings.n_classes)