2 Star 2 Fork 1

openvinotoolkit-prc / nncf

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mobilenet_v2_imagenet_mixed_int_autoq_staged.json 1.23 KB
一键复制 编辑 原始数据 按行查看 历史
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,
"epochs": 10,
"workers": 6,
"multiprocessing_distributed": true,
"target_device": "NPU",
"compression": {
"algorithm": "quantization",
"initializer": {
"batchnorm_adaptation": {
"num_bn_adaptation_samples": 4000
},
"range":
{
"type": "mean_min_max",
"num_init_samples": 1500
},
"precision": {
"type": "autoq",
"bits": [2, 4, 8],
"iter_number": 300,
"compression_ratio": 0.15,
"eval_subset_ratio": 0.20,
"dump_init_precision_data": true
}
},
"params": {
"base_lr": 3.1e-4,
"base_wd": 1e-5,
"batch_multiplier": 1,
"activations_quant_start_epoch": 0,
"weights_quant_start_epoch": 0,
"lr_poly_drop_start_epoch": 0,
"lr_poly_drop_duration_epochs": 10
}
},
"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

搜索帮助

53164aa7 5694891 3bd8fe86 5694891