2 Star 2 Fork 1

openvinotoolkit-prc / nncf

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
resnet50_imagenet_magnitude_sparsity_accuracy_aware.json 1.26 KB
一键复制 编辑 原始数据 按行查看 历史
Evgeny Tsykunov 提交于 2021-09-06 18:41 . update config (#920)
{
"model": "ResNet50",
"pretrained": true,
"input_info": {
"sample_size": [1, 224, 224, 3]
},
"batch_size": 320,
"epochs": 65,
"optimizer": {
"type": "sgd",
"schedule_type": "multistep",
"schedule_params": {
"base_lr": 0.01,
"steps": [20, 40, 60],
"gamma": 0.1
},
"optimizer_params": {
"momentum": 0.9,
"nesterov": true
}
},
"dataset": "imagenet2012",
"dataset_type": "tfds",
"accuracy_aware_training": {
"mode": "adaptive_compression_level",
"params": {
"maximal_relative_accuracy_degradation": 1.0,
"initial_training_phase_epochs": 100,
"patience_epochs": 40
}
},
"compression": {
"algorithm": "magnitude_sparsity",
"sparsity_init": 0.1,
"params": {
"schedule": "multistep",
"multistep_steps": [
5,
10,
15,
20,
25
],
"multistep_sparsity_levels": [
0.1,
0.15,
0.20,
0.25,
0.30,
0.35
]
}
}
}
Python
1
https://gitee.com/openvinotoolkit-prc/nncf.git
git@gitee.com:openvinotoolkit-prc/nncf.git
openvinotoolkit-prc
nncf
nncf
develop

搜索帮助

53164aa7 5694891 3bd8fe86 5694891