optimum/exporters/onnx/model_configs.py @@ -550,6 +550,11 @@ def inputs(self) -> Dict[str, Dict[int, str]]: return {"pixel_values": {0: "batch_size", 1: "num_channels", 2: "height", 3: "width"}} class CvTOnnxConfig(ViTOnnxConfig): DEFAULT_ONNX_OPSET = 13 ATOL...
# global quantize parameterbatch_num : 2 activation_offset :trueskip_layers :"layer1.1.conv2"skip_layer_types:"Conv"do_fusion:trueskip_fusion_layers :"layer1.1.conv2"common_config : { arq_quantize : { channel_wise :true} ifmr_quantize : { search_range_start : 0....
调用示例 from msmodelslim.onnx.post_training_quant import QuantConfig def custom_read_data(): calib_data = [] # TODO 读取数据集,进行数据预处理,将数据存入calib_data return calib_data calib_data = custom_read_data() quant_config = QuantConfig(calib_data=calib_data, amp_num=5)暂...
onnxruntime orttraining rust samples tools winml .clang-format .clang-tidy .dockerignore .gitattributes .gitignore .gitmodules .lintrunner.toml CITATION.cff CODEOWNERS CONTRIBUTING.md LICENSE NuGet.config ORT_icon_for_light_bg.png README.md SECURITY.md ThirdPartyNotices.txt VERSION_NUMBER build....
文件 main 该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。 克隆/下载 git config --global user.name userName git config --global user.email userEmail main onnx-community-whisper-small.en / config.json config.json2.15 KB ...
在您的案例中,问题可能出在 onnx_intra_nth 的赋值或使用上。下面我将按照您提供的提示逐一分析并给出解决方案: 1. 确认 sess_config.intra_op_num_threads 和onnx_intra_nth 的数据类型 sess_config.intra_op_num_threads 通常期望一个整数(int)类型,用于设置TensorFlow会话(或类似环境)中的线程数。 onnx_...
- repo: https://github.com/pre-commit/pre-commit-hooks rev: v4.6.0 hooks: - id: check-merge-conflict - id: check-symlinks - id: end-of-file-fixer - id: trailing-whitespace - id: detect-private-key - id: check-symlinks - id: check-added-large-files - repo: local hooks: - id...
Breadcrumbs onnx-tool /data /public / config.pyTop File metadata and controls Code Blame 334 lines (324 loc) · 9.3 KB Raw import numpy from onnx_tool import create_ndarray_f32, create_ndarray_int64 public_models = { 'folder': 'data/public', 'models': [ { 'name': 'so-vits-svc...
amp_num=0, keep_acc=None, sigma=25, disable_first_layer=True, disable_last_layer=True, is_optimize_graph=True, is_dynamic_shape=False, use_onnx=True, num_input=0, quant_param_ops=None, atc_input_shape=None, graph_optimize_level=0, shut_down_structures=None, device_id=0, o...
Remove xfails from in-tree onnx op test config. This way, we won't need to update the xfails in this repository after nightly releases affect which tests are passing. Downstream (iree-org/iree) will still use the xfail behavior, while this repository will just report the current status,...