torch.onnx.export 无尺度信息
时间: 2024-08-14 13:09:45 浏览: 44
pytorch 1.9.0 torch.onnx.export导出jit script模型报错
`torch.onnx.export` 是 PyTorch 中用于将模型导出为 ONNX (Open Neural Network Exchange) 格式的一个函数。ONNX 是一种跨框架的模型交换格式,但默认情况下,它并不包含原始数据的尺度信息,如图片的大小或输入张量的量化级别等。
当你使用 `torch.onnx.export` 导出模型时,如果模型对输入有特定的预处理步骤(例如,归一化到某个范围),这些细节不会直接反映在 ONNX 文件中。ONNX 模型通常只保存网络结构、节点及其权重,而不保留训练或应用时的具体数值上下文。
如果你需要保存这些额外的信息,比如像素值的范围,一般是在模型训练或转换过程中记录,并在后续运行时通过元数据或者其他形式附加到模型上。在实际部署中,你需要提供一个预处理步骤,以便复现模型在训练时的输入处理。
阅读全文