如何把字典存入.pt文件

你可以使用Python中的`torch.save()`方法将字典存储为.pt文件。以下是一个例子：

import torch

# 定义要保存的字典
my_dict = {'key1': 'value1', 'key2': 'value2'}

# 定义保存路径和文件名
save_path = 'my_dict.pt'

# 将字典保存为.pt文件
torch.save(my_dict, save_path)

这将把`my_dict`字典存储为`my_dict.pt`文件。当你需要使用这个字典时，可以使用`torch.load()`方法来加载.pt文件并恢复字典：

# 从.pt文件中加载字典
loaded_dict = torch.load(save_path)

# 打印字典
print(loaded_dict)

这将输出：

{'key1': 'value1', 'key2': 'value2'}