2 Star 2 Fork 1

openvinotoolkit-prc / nncf

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mobilenet_v2_imagenet_mixed_int_hawq.json 901 Bytes
一键复制 编辑 原始数据 按行查看 历史
Nikita Malinin 提交于 2024-02-21 11:22 . Changed VPU to NPU device (#2488)
{
"model": "mobilenet_v2",
"pretrained": true,
"input_info": {
"sample_size": [
2,
3,
224,
224
]
},
"num_classes": 1000,
"batch_size": 200,
"workers": 6,
"multiprocessing_distributed": true,
"optimizer": {
"base_lr": 3.1e-4,
"schedule_type": "plateau",
"type": "Adam",
"schedule_params": {
"threshold": 0.1,
"cooldown": 3
},
"weight_decay": 1e-05
},
"target_device": "NPU",
"compression": {
"algorithm": "quantization",
"quantize_inputs": true,
"initializer": {
"precision": {
"type": "hawq",
"bits": [4, 8],
"compression_ratio": 1.5
}
}
},
"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

搜索帮助