2 Star 2 Fork 1

openvinotoolkit-prc / nncf

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
resnet18_imagenet_pruning_accuracy_aware.json 1.07 KB
一键复制 编辑 原始数据 按行查看 历史
{
"model": "resnet18",
"pretrained": true,
"batch_size": 512,
"epochs": 100,
"input_info": {
"sample_size": [
1,
3,
224,
224
]
},
"optimizer": {
"type": "SGD",
"base_lr": 0.1,
"weight_decay": 1e-4,
"schedule_type": "multistep",
"steps": [
20,
40,
60,
80
],
"optimizer_params": {
"momentum": 0.9,
"nesterov": true
}
},
"accuracy_aware_training": {
"mode": "adaptive_compression_level",
"params": {
"maximal_relative_accuracy_degradation": 1.0,
"initial_training_phase_epochs": 100,
"patience_epochs": 100
}
},
"compression": {
"algorithm": "filter_pruning",
"pruning_init": 0.1,
"params": {
"schedule": "exponential",
"pruning_target": 0.4,
"pruning_steps": 15,
"filter_importance": "geometric_median"
}
}
}
Python
1
https://gitee.com/openvinotoolkit-prc/nncf.git
git@gitee.com:openvinotoolkit-prc/nncf.git
openvinotoolkit-prc
nncf
nncf
develop

搜索帮助