主页 / coding / 问题 / 79378157
Accepted
Mickey Han
Asked: 2025-01-22 22:35:06 +0800 CST

HuggingFace 模型 - OnnxRuntime - Jupyter Notebook 打印模型摘要

  • 772

非常感谢您阅读我的问题，如果这是一个显而易见的问题，请原谅。

我使用 anaconda navigator：通过管道安装来自 OpenAi 的模型 whisper，这是一个音频到文本转换模型，我使用 jupyter notebook，当我刚运行模型的单元时，会出现这个模块摘要，这对于了解模型是什么非常有用：

在此处输入图片描述

但是，使用另一个 pip 安装的模型： https://huggingface.co/breezedeus/pix2text-mfr 我注意到区别在于它是最佳的。onnxruntime

在此处输入图片描述

当我做与上面相同的事情时，它会返回一个内存位置？或者是？

抱歉，如果这是一个简单的问题，我尝试谷歌搜索了一下，但不知道要搜索什么关键词 - “onnx pytorch 模型摘要”？有没有办法得到如上所述的模型摘要？

非常感谢您阅读我的问题。

pytorch

1 个回答

  1. Best Answer

    ORT 模型输出只是表示 Python 中对象的默认字符串，提供类名和内存地址。它们都是 Python 中的有效对象，但第一个模型会覆盖__str__()打印模型时显示图层的方法。

    ORT 模型没有与__str__()whisper 模型相同的方法（例如 override），因为它不是基于 PyTorch 的模型。相反，它使用 ONNX 图定义和运算符。您可以使用 Netron 之类的工具。您可以尝试打印编码器和解码器图，但可读性不强。

    import onnx

# Encoder model
encoder_onnx_model = onnx.load(model.encoder_model_path)
print(onnx.helper.printable_graph(encoder_onnx_model.graph))

# Decoder model
decoder_onnx_model = onnx.load(model.decoder_model_path)
print(onnx.helper.printable_graph(decoder_onnx_model.graph))
    • 1

相关问题