BimAnt › 在线工具 › PyTorch 模型在线查看器

PyTorch 模型在线查看

把 TorchScript 导出的 .pt / .pth 拖进 Netron,即可查看计算图;普通权重文件请先导出 TorchScript 或 ONNX。

打开 Netron 神经网络模型在线查看器 →

怎么用

  1. 在 PyTorch 中导出 TorchScript:torch.jit.trace(model, example) 或 torch.jit.script(model),保存为 .pt。
  2. 把导出的 .pt / .pth 拖进页面。
  3. 查看计算图与算子结构。
  4. 也可以先导出 ONNX,再用本工具打开。

支持的能力

常见问题

为什么我的 .pt 打不开?

普通训练存档(仅 state_dict)不含计算图,无法可视化;请先用 torch.jit.trace / script 导出 TorchScript,或导出为 ONNX。

TorchScript 和普通 .pt 怎么区分?

TorchScript 文件由 torch.jit 导出、包含 root module;若提示“File does not contain root module or state dictionary”,说明不是 TorchScript 格式。

直接打开 pickle 格式的权重安全吗?

pickle 反序列化有代码执行风险;建议优先使用 TorchScript 或 ONNX 这类安全格式。

相关工具

Make Sense 在线图像标注 ONNX 模型库 AI Model Zoo