如何区分并理解PyTorch中模型文件.pt, .pth, .pkl的保存差异？

本文共计1236个文字，预计阅读时间需要5分钟。

我们经常会看到文件后缀名为.pt、.pth和.pkl的PyTorch模型文件。这些文件在格式上有什么区别吗？

实际上，这些文件在格式上并没有本质的区别，它们的区别仅仅在于后缀名不同。

- .pt文件：这是PyTorch默认的模型保存格式，它使用HDF5文件格式存储模型。- .pth文件：这是PyTorch早期版本使用的格式，也是基于HDF5，但它是一种更为紧凑的存储方式。- .pkl文件：这是Python的pickle模块使用的格式，可以存储任意Python对象，包括PyTorch模型。

在使用torch.save()函数保存模型时，默认会保存为.pt格式。不过，你也可以通过设置参数来自定义文件的后缀名。所以，这些文件的区别主要在于后缀名，而它们的实际内容是相同的。

我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件，这几种模型文件在格式上有什么区别吗？

其实它们并不是在格式上有区别，只是后缀不同而已（仅此而已），在用torch.save()函数保存模型文件时，各人有不同的喜好，有些人喜欢用.pt后缀，有些人喜欢用.pth或.pkl.用相同的torch.save（）语句保存出来的模型文件没有什么不同。

在pytorch官方的文档/代码里，有用.pt的，也有用.pth的。一般惯例是使用.pth,但是官方文档里貌似.pt更多，而且官方也不是很在意固定用一种。