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)
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()
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)