2 Star 2 Fork 1

openvinotoolkit-prc / nncf

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
resnet50_imagenet_mixed_int_hawq.json 995 Bytes
一键复制 编辑 原始数据 按行查看 历史
Nikita Malinin 提交于 2024-02-21 11:22 . Changed VPU to NPU device (#2488)
{
"model": "resnet50",
"pretrained": true,
"input_info": {
"sample_size": [
1,
3,
224,
224
]
},
"num_classes": 1000,
"batch_size": 150,
"workers": 6,
"multiprocessing_distributed": true,
"epochs": 30,
"target_device": "NPU",
"compression": {
"algorithm": "quantization",
"initializer": {
"precision": {
"type": "hawq",
"bits": [4, 8],
"compression_ratio": 1.4
}
},
"params": {
"base_lr": 3.1e-4,
"base_wd": 1e-5,
"batch_multiplier": 1,
"activations_quant_start_epoch": 3,
"weights_quant_start_epoch": 12,
"lr_poly_drop_start_epoch": 20,
"lr_poly_drop_duration_epochs": 10,
"disable_wd_start_epoch": 20
}
},
"no_strip_on_export": true,
"export_to_ir_via_onnx": true
}
Python
1
https://gitee.com/openvinotoolkit-prc/nncf.git
git@gitee.com:openvinotoolkit-prc/nncf.git
openvinotoolkit-prc
nncf
nncf
develop

搜索帮助