# -*- coding: utf-8 -*-
"""Configurations for network slimming + L2 magnitude pruning.

- Author: <NAME>
- Email: <EMAIL>
"""

from config.train.cifar100 import densenet_small

train_config = densenet_small.config
train_config.update({"REGULARIZER": "BnWeight", "REGULARIZER_PARAMS": dict(coeff=1e-5)})
config = {
    "TRAIN_CONFIG": train_config,
    "N_PRUNING_ITER": 15,
    "PRUNE_METHOD": "SlimMagnitude",
    "PRUNE_PARAMS": dict(
        PRUNE_AMOUNT=0.2,
        NORM=2,
        STORE_PARAM_BEFORE=train_config["EPOCHS"],
        TRAIN_START_FROM=0,
        PRUNE_AT_BEST=False,
    ),
}

# -*- coding: utf-8 -*-
"""Configurations for network slimming + L2 magnitude pruning.

- Author: <NAME>
- Email: <EMAIL>
"""

from config.train.cifar100 import densenet_small

train_config = densenet_small.config
train_config.update({"REGULARIZER": "BnWeight", "REGULARIZER_PARAMS": dict(coeff=1e-5)})
config = {
    "TRAIN_CONFIG": train_config,
    "N_PRUNING_ITER": 15,
    "PRUNE_METHOD": "SlimMagnitude",
    "PRUNE_PARAMS": dict(
        PRUNE_AMOUNT=0.2,
        NORM=2,
        STORE_PARAM_BEFORE=train_config["EPOCHS"],
        TRAIN_START_FROM=0,
        PRUNE_AT_BEST=False,
    ),
}

# -*- coding: utf-8 -*- Configurations for network slimming + L2 magnitude pruning.  - Author: <NAME> - Email: <EMAIL>