请注意,本文内容源自机器翻译,可能存在语法或其它翻译错误,仅供参考。如需获取准确内容,请参阅链接中的英语原文或自行翻译。
器件型号:CODECOMPOSER 工具与软件:
您好、我正在使用我们公司的示例代码进行自动化识别训练(QAT)、我在代码中遇到一个问题:
我看到贵公司提供的量化工具:Github 中的 Readme 里面提到:
可选:我们提供了一个名为 torchvision.edgeailite.xnn.utils.load_weights ()的实用函数、它可以打印哪些参数已正确加载以及哪些参数未正确加载-如果需要、您可以使用此加载函数来确保您的参数已正确加载。
我想使用 torchvision.edgeailite.xnn.utils.load_weights ()工具代码来进行量化训练后的模型检查、但是、我看到示例代码给出如下代码:
# load pretrained model if pretrained_data is not None and not is_onnx_model: model_orig = get_model_orig(model) for (p_data,p_file) in zip(pretrained_data, pretrained_files): print("=> using pretrained weights from: {}".format(p_file)) if hasattr(model_orig, 'load_weights'): model_orig.load_weights(pretrained=p_data, change_names_dict=change_names_dict) else: xnn.utils.load_weights(get_model_orig(model), pretrained=p_data, change_names_dict=change_names_dict)
通过以上代码、我无法直接执行else中的 xnn。实用程序。load_weights()函数。
请问我需要修改其他的代码或参数吗?或者删除如果、否则、直接执行xnn。实用程序。load_weights()函数?
期待您的回复!μ A