.pickle文件和.pt文件

import pickle
data = {'name': 'Alice', 'age': 30}
# 保存为.pickle文件
with open('data.pickle', 'wb') as file:
    pickle.dump(data, file)

• 加载方法：使用Python的pickle模块可以加载.pickle文件。下面是一个示例代码：

  import pickle
  # 加载.pickle文件
  with open('data.pickle', 'rb') as file:
      data = pickle.load(file)
  # 使用加载的数据
  print(data)
  

  1. .pt文件：

  • 作用和意义：.pt文件通常用于存储PyTorch模型、张量等数据，是PyTorch框架自定义的文件格式。
  • 保存方法：使用PyTorch框架的相关函数可以保存数据为.pt文件。下面是一个示例代码：

    import torch
    data = torch.tensor([1, 2, 3])
    # 保存为.pt文件
    torch.save(data, 'data.pt')
    

    • 加载方法：使用PyTorch框架的相关函数可以加载.pt文件。下面是一个示例代码：

      import torch
      # 加载.pt文件
      data = torch.load('data.pt')
      # 使用加载的数据
      print(data)
      

      总结：.pickle文件适用于一般的Python对象的序列化和反序列化，而.pt文件则适用于PyTorch相关数据（如模型、张量）的存储和加载。

      （）（）