1 Star 2 Fork 1

openvinotoolkit-prc / nncf_pytorch

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
resnet18_cifar10_accuracy_aware.json 1.21 KB
一键复制 编辑 原始数据 按行查看 历史
{
"model": "resnet18",
"weights": "https://storage.openvinotoolkit.org/repositories/nncf/examples/torch/classification/resnet18_cifar10.pth",
"batch_size": 512,
"input_info": {
"sample_size": [
1,
3,
32,
32
]
},
"dataset": "cifar10",
"num_classes": 10,
"optimizer": {
"type": "SGD",
"base_lr": 0.001,
"weight_decay": 1e-4,
"schedule_type": "multistep",
"steps": [
60,
100
],
"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": 110,
"patience_epochs": 110,
"maximal_total_epochs": 1000
}
},
"compression": {
"algorithm": "filter_pruning",
"pruning_init": 0.1,
"params": {
"schedule": "exponential",
"pruning_target": 0.4,
"pruning_steps": 20,
"filter_importance": "geometric_median"
}
}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/openvinotoolkit-prc/nncf_pytorch.git
git@gitee.com:openvinotoolkit-prc/nncf_pytorch.git
openvinotoolkit-prc
nncf_pytorch
nncf_pytorch
develop

搜索帮助

344bd9b3 5694891 D2dac590 5694891