Skip to main content
 首页 » 编程设计

python之PyTorch 中的 .pt、.pth 和 .pwf 扩展名有什么区别

2025年05月04日20daizhj

我在一些代码示例中看到,人们使用 .pwf 作为模型文件保存格式。但在 PyTorch 文档中,建议使用 .pt 和 .pth。我使用 .pwf 并且在小型 1->16->16 卷积网络中工作得很好。

我的问题是这些格式之间有什么区别?为什么 PyTorch 文档中甚至不推荐 .pwf 扩展名,为什么人们仍在使用它?

请您参考如下方法:

列出的扩展名之间没有区别:.pt、.pth、.pwf。可以使用他想要的任何扩展名。因此,如果您使用 torch.save() 来保存模型,那么它默认使用 python pickle (pickle_module=pickle) 来保存对象和一些元数据。因此,您可以自由选择所需的扩展,只要它不会与任何其他标准化扩展发生冲突。

话虽如此,却是not recommended to use .pth extension当检查点模型时,因为它与 Python path (.pth) configuration files 发生冲突.因此,我自己使用 .pth.tar 或 .pt 但不使用 .pth 或任何其他扩展。


PyTorch 中检查点模型的标准方法尚未最终确定。在撰写本文时,这是一个悬而未决的问题:Recommend a different file extension for models (.PTH is a special extension for Python) - issues/14864

已经suggested by @soumith使用:

  • .pt 用于 pickle 格式的检查点模型
  • .ptc 用于 pytorch 编译的检查点模型(用于 JIT)