我在一些代码示例中看到,人们使用.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
用于腌制格式的检查点模型.ptc
用于在pytorch中编译检查点模型(用于JIT)
关于python - PyTorch中的.pt,.pth和.pwf扩展之间有什么区别?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/59095824/