from config import PruningConfig
from configs.cnn_chars74k_aging import training_config, bound_config, search_config, search_algorithm
training_config.pruning = PruningConfig(structured=True,
start_pruning_at_epoch=20,
finish_pruning_by_epoch=53,
min_sparsity=0.1,
max_sparsity=0.85)
from config import PruningConfig
from configs.cnn_fashion_aging import training_config, bound_config, search_config, search_algorithm
training_config.pruning = PruningConfig(structured=True,
start_pruning_at_epoch=3,
finish_pruning_by_epoch=38,
min_sparsity=0.05,
max_sparsity=0.90)
from config import PruningConfig, BoundConfig
from configs.cnn_mnist_aging import training_config, search_config, search_algorithm
training_config.pruning = PruningConfig(structured=False,
start_pruning_at_epoch=3,
finish_pruning_by_epoch=18,
min_sparsity=0.2,
max_sparsity=0.98)
bound_config = BoundConfig(error_bound=0.025,
peak_mem_bound=None,
model_size_bound=1000,
mac_bound=None)