[原创] PyTorch模型 .pt，.pth，.pkl 的区别

我们经常会看到后缀名为 .pt，.pth，.pkl 的PyTorch模型文件，这几种模型文件在格式上有什么区别吗？
其实它们并不是在格式上有区别，而只是后缀上不同而已（仅此而已）。在用 torch.save() 函数保存模型文件的时候，各人有不同的喜好，有些人喜欢用 .pt 后缀，有些人喜欢用 .pth 或 .pkl。用相同的 torch.save() 语句保存出来的模型文件没有什么不同。
在PyTorch官方的文档/代码里，有用 .pt 的，也有用 .pth 的。
据某些文章的说法，一般惯例是使用 .pth，但是官方文档里貌似 .pt 更多，而且官方也不是很在意固定用一种，大家就自便吧。

另外，为什么会有 .pkl 这种后缀名？因为 Python 有一个序列化/反序列化的模块，名字叫 pickle，用它保存的文件，通常会起一个 .pkl 的后缀名。torch.save() 正是使用了Python pickle来保存模型的，因此使用 .pkl 作为模型文件的后缀也就不奇怪了。
文章来源：https://www.codelast.com/
➤➤ 版权声明 ➤➤ 
转载需注明出处：codelast.com 
感谢关注我的微信公众号（微信扫一扫）：